diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-07-09 16:55:37 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-09 17:02:35 -0700 |
| commit | e0541b714d0df485a79ece616575ebb2a71b818c (patch) | |
| tree | 6e32dbbdbb6ccb254af057d97900b7e158f3eff9 /tensorflow/compiler/tests/BUILD | |
| parent | 3493816ae9bf83e6a1a7639ebce3a05b944e2cd2 (diff) | |
[TF:XLA] Add implementation of ResourceApplyAdagradDA and ResourceApplyAdaMax.
PiperOrigin-RevId: 203855534
Diffstat (limited to 'tensorflow/compiler/tests/BUILD')
| -rw-r--r-- | tensorflow/compiler/tests/BUILD | 27 |
1 files changed, 27 insertions, 0 deletions
diff --git a/tensorflow/compiler/tests/BUILD b/tensorflow/compiler/tests/BUILD index 69ff0d99cb..8b25147899 100644 --- a/tensorflow/compiler/tests/BUILD +++ b/tensorflow/compiler/tests/BUILD @@ -98,6 +98,19 @@ tf_xla_py_test( ) tf_xla_py_test( + name = "adagrad_da_test", + size = "small", + srcs = ["adagrad_da_test.py"], + deps = [ + ":xla_test", + "//tensorflow/python:array_ops", + "//tensorflow/python:framework", + "//tensorflow/python:platform_test", + "//tensorflow/python:training", + ], +) + +tf_xla_py_test( name = "adam_test", size = "small", srcs = ["adam_test.py"], @@ -112,6 +125,20 @@ tf_xla_py_test( ) tf_xla_py_test( + name = "adamax_test", + size = "small", + srcs = ["adamax_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 = "addsign_test", size = "small", srcs = ["addsign_test.py"], |