diff options
author | 2018-07-06 18:02:00 -0700 | |
---|---|---|
committer | 2018-07-07 21:21:42 -0700 | |
commit | d7416d53ef09052a3dc10ec54ee5bd19e883cf86 (patch) | |
tree | 7bf2b5cbee0ce8ad768da5116fdce07979b26982 /tensorflow/compiler/tests/BUILD | |
parent | 1caaea99e0156523e7d65f7e54cc2ac7117a8f90 (diff) |
[TF:XLA] Add implementation of ResourceApplyPowerSign and ResourceApplyAddSign.
PiperOrigin-RevId: 203547001
Diffstat (limited to 'tensorflow/compiler/tests/BUILD')
-rw-r--r-- | tensorflow/compiler/tests/BUILD | 28 |
1 files changed, 28 insertions, 0 deletions
diff --git a/tensorflow/compiler/tests/BUILD b/tensorflow/compiler/tests/BUILD index 3be391ba90..69ff0d99cb 100644 --- a/tensorflow/compiler/tests/BUILD +++ b/tensorflow/compiler/tests/BUILD @@ -112,6 +112,34 @@ tf_xla_py_test( ) tf_xla_py_test( + name = "addsign_test", + size = "small", + srcs = ["addsign_test.py"], + deps = [ + ":xla_test", + "//tensorflow/contrib/opt:opt_py", + "//tensorflow/python:array_ops", + "//tensorflow/python:client_testlib", + "//tensorflow/python:framework", + "//tensorflow/python:training", + ], +) + +tf_xla_py_test( + name = "powersign_test", + size = "small", + srcs = ["powersign_test.py"], + deps = [ + ":xla_test", + "//tensorflow/contrib/opt:opt_py", + "//tensorflow/python:array_ops", + "//tensorflow/python:client_testlib", + "//tensorflow/python:framework", + "//tensorflow/python:training", + ], +) + +tf_xla_py_test( name = "argminmax_test", size = "small", srcs = ["argminmax_test.py"], |