diff options
author | 2018-07-03 13:10:12 -0700 | |
---|---|---|
committer | 2018-07-03 13:16:25 -0700 | |
commit | 02ed358a986496e387d5f2e52865b10606e52c0a (patch) | |
tree | e5884e91843513cf38c570b63e94eaf06b75ad64 /tensorflow/compiler/tests/BUILD | |
parent | 694d2e708e34534407da6798a50cf64cfd7f6c38 (diff) |
[TF:XLA] Add implementation of ResourceApplyProximalGradientDescent and ResourceApplyProximalAdagrad.
PiperOrigin-RevId: 203171693
Diffstat (limited to 'tensorflow/compiler/tests/BUILD')
-rw-r--r-- | tensorflow/compiler/tests/BUILD | 26 |
1 files changed, 26 insertions, 0 deletions
diff --git a/tensorflow/compiler/tests/BUILD b/tensorflow/compiler/tests/BUILD index 080b1c9c35..273641f197 100644 --- a/tensorflow/compiler/tests/BUILD +++ b/tensorflow/compiler/tests/BUILD @@ -584,6 +584,32 @@ tf_xla_py_test( ) tf_xla_py_test( + name = "proximal_adagrad_test", + size = "medium", + srcs = ["proximal_adagrad_test.py"], + deps = [ + ":xla_test", + "//tensorflow/python:array_ops", + "//tensorflow/python:client_testlib", + "//tensorflow/python:framework", + "//tensorflow/python:training", + ], +) + +tf_xla_py_test( + name = "proximal_gradient_descent_test", + size = "medium", + srcs = ["proximal_gradient_descent_test.py"], + deps = [ + ":xla_test", + "//tensorflow/python:array_ops", + "//tensorflow/python:client_testlib", + "//tensorflow/python:framework", + "//tensorflow/python:training", + ], +) + +tf_xla_py_test( name = "random_ops_test", size = "small", srcs = ["random_ops_test.py"], |