diff options
author | 2018-06-14 12:05:53 -0700 | |
---|---|---|
committer | 2018-06-14 12:11:47 -0700 | |
commit | d943de372a989ca6bc44058e35ba9f26591b42b4 (patch) | |
tree | c9d30ee4a20abf0f78bb2dfb5c326d7c0b076674 /tensorflow/python/kernel_tests/distributions/BUILD | |
parent | 7ccf1937b863a7f5cfb5d159d44671138d7393bf (diff) |
Support non-static shape in `tf.distributions.Categorical`.
PiperOrigin-RevId: 200596358
Diffstat (limited to 'tensorflow/python/kernel_tests/distributions/BUILD')
-rw-r--r-- | tensorflow/python/kernel_tests/distributions/BUILD | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/python/kernel_tests/distributions/BUILD b/tensorflow/python/kernel_tests/distributions/BUILD index cf2e8832fd..985922245e 100644 --- a/tensorflow/python/kernel_tests/distributions/BUILD +++ b/tensorflow/python/kernel_tests/distributions/BUILD @@ -93,6 +93,7 @@ cuda_py_test( size = "small", srcs = ["categorical_test.py"], additional_deps = [ + "@absl_py//absl/testing:parameterized", "//tensorflow/python/ops/distributions", "//third_party/py/numpy", "//tensorflow/python:array_ops", |