diff options
author | 2017-11-14 14:41:12 -0800 | |
---|---|---|
committer | 2017-11-14 14:52:30 -0800 | |
commit | b8054c19b7d72cfb7eb07552a8d0385ffd8810d7 (patch) | |
tree | dfd57ff6078992c06e0bd01ac52b913af715a7ff /tensorflow/contrib/bayesflow/BUILD | |
parent | e8b2049f5be7accae9f272972acfc5afb36c5ef2 (diff) |
Add `tf.contrib.bayesflow.layers`, a collection of probabilistic (neural) layers.
PiperOrigin-RevId: 175737756
Diffstat (limited to 'tensorflow/contrib/bayesflow/BUILD')
-rw-r--r-- | tensorflow/contrib/bayesflow/BUILD | 21 |
1 files changed, 20 insertions, 1 deletions
diff --git a/tensorflow/contrib/bayesflow/BUILD b/tensorflow/contrib/bayesflow/BUILD index 9f3650e8f9..a262d4aecd 100644 --- a/tensorflow/contrib/bayesflow/BUILD +++ b/tensorflow/contrib/bayesflow/BUILD @@ -19,6 +19,7 @@ py_library( srcs = ["__init__.py"] + glob(["python/ops/*.py"]), srcs_version = "PY2AND3", deps = [ + "//tensorflow/contrib/distributions:distributions_py", "//tensorflow/contrib/framework:framework_py", "//tensorflow/python:array_ops", "//tensorflow/python:control_flow_ops", @@ -32,7 +33,6 @@ py_library( "//tensorflow/python:random_ops", "//tensorflow/python:state_ops", "//tensorflow/python:util", - "//tensorflow/python/ops/distributions", "//third_party/py/numpy", ], ) @@ -100,6 +100,25 @@ cuda_py_test( ) cuda_py_test( + name = "layers_dense_variational_test", + size = "small", + srcs = ["python/kernel_tests/layers_dense_variational_test.py"], + additional_deps = [ + ":bayesflow_py", + "//third_party/py/numpy", + "//tensorflow/contrib/distributions:distributions_py", + "//tensorflow/python/ops/distributions", + "//tensorflow/python:array_ops", + "//tensorflow/python:client_testlib", + "//tensorflow/python:framework_for_generated_wrappers", + "//tensorflow/python:gradients", + "//tensorflow/python:linalg_ops", + "//tensorflow/python:math_ops", + "//tensorflow/python:nn_ops", + ], +) + +cuda_py_test( name = "monte_carlo_test", size = "small", srcs = ["python/kernel_tests/monte_carlo_test.py"], |