diff options
Diffstat (limited to 'tensorflow/contrib/opt/BUILD')
-rw-r--r-- | tensorflow/contrib/opt/BUILD | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/tensorflow/contrib/opt/BUILD b/tensorflow/contrib/opt/BUILD index 096d2270e4..022e5ab06f 100644 --- a/tensorflow/contrib/opt/BUILD +++ b/tensorflow/contrib/opt/BUILD @@ -18,6 +18,7 @@ py_library( "python/training/external_optimizer.py", "python/training/lazy_adam_optimizer.py", "python/training/moving_average_optimizer.py", + "python/training/multitask_optimizer_wrapper.py", "python/training/nadam_optimizer.py", "python/training/variable_clipping_optimizer.py", ], @@ -96,6 +97,23 @@ py_test( ) py_test( + name = "multitask_optimizer_wrapper_test", + srcs = ["python/training/multitask_optimizer_wrapper_test.py"], + srcs_version = "PY2AND3", + deps = [ + ":opt_py", + "//tensorflow/python:client", + "//tensorflow/python:client_testlib", + "//tensorflow/python:constant_op", + "//tensorflow/python:dtypes", + "//tensorflow/python:training", + "//tensorflow/python:variables", + "//third_party/py/numpy", + "@six_archive//:six", + ], +) + +py_test( name = "lazy_adam_optimizer_test", srcs = ["python/training/lazy_adam_optimizer_test.py"], srcs_version = "PY2AND3", |