diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2017-12-20 10:31:55 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-12-20 12:08:36 -0800 |
commit | db67415c8fae6205dff013b8eb5cf75e1764f9f2 (patch) | |
tree | 759f57321957b1205f2554c05cd98c5254ef8fd3 /tensorflow/contrib/bayesflow/BUILD | |
parent | dbedc1312061ce1437a6c71d279675e4e2a79595 (diff) |
Adding an optimizer module for ConstantSGD.
PiperOrigin-RevId: 179703560
Diffstat (limited to 'tensorflow/contrib/bayesflow/BUILD')
-rw-r--r-- | tensorflow/contrib/bayesflow/BUILD | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/tensorflow/contrib/bayesflow/BUILD b/tensorflow/contrib/bayesflow/BUILD index 4e0520fa33..aaae904fe1 100644 --- a/tensorflow/contrib/bayesflow/BUILD +++ b/tensorflow/contrib/bayesflow/BUILD @@ -219,6 +219,27 @@ cuda_py_test( ], ) +cuda_py_test( + name = "variational_sgd_optimizer_test", + size = "small", + srcs = ["python/kernel_tests/variational_sgd_optimizer_test.py"], + additional_deps = [ + ":bayesflow_py", + "//third_party/py/numpy", + "//tensorflow/contrib/distributions:distributions_py", + "//tensorflow/contrib/layers:layers_py", + "//tensorflow/python/ops/distributions", + "//tensorflow/python:client_testlib", + "//tensorflow/python:framework", + "//tensorflow/python:framework_for_generated_wrappers", + "//tensorflow/python:framework_test_lib", + "//tensorflow/python:gradients", + "//tensorflow/python:math_ops", + "//tensorflow/python:platform_test", + "//tensorflow/python:random_seed", + ], +) + filegroup( name = "all_files", srcs = glob( |