diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-06-07 11:45:01 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-07 11:47:20 -0700 |
commit | 642dc96bd4627a4f6305cf61b8553324054d9122 (patch) | |
tree | 49bc12899ff6ca0e631746b69cd91f34f8f52366 /tensorflow/contrib/distributions/BUILD | |
parent | e343b8072833765c85a5685b0f56b1b3d6add275 (diff) |
Add FillTriangular Bijector to create triangular matrices.
PiperOrigin-RevId: 199670547
Diffstat (limited to 'tensorflow/contrib/distributions/BUILD')
-rw-r--r-- | tensorflow/contrib/distributions/BUILD | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/tensorflow/contrib/distributions/BUILD b/tensorflow/contrib/distributions/BUILD index 23d9dbcd91..d8baf49e81 100644 --- a/tensorflow/contrib/distributions/BUILD +++ b/tensorflow/contrib/distributions/BUILD @@ -941,6 +941,25 @@ cuda_py_test( ) cuda_py_test( + name = "fill_triangular_test", + size = "small", + srcs = ["python/kernel_tests/bijectors/fill_triangular_test.py"], + additional_deps = [ + ":bijectors_py", + ":distributions_py", + "//third_party/py/numpy", + "@six_archive//:six", + "//tensorflow/contrib/linalg:linalg_py", + "//tensorflow/python:array_ops", + "//tensorflow/python:client_testlib", + "//tensorflow/python:framework_for_generated_wrappers", + "//tensorflow/python:framework_test_lib", + "//tensorflow/python:math_ops", + "//tensorflow/python:platform_test", + ], +) + +cuda_py_test( name = "gumbel_test", size = "small", srcs = ["python/kernel_tests/bijectors/gumbel_test.py"], |