diff options
Diffstat (limited to 'tensorflow/python/kernel_tests/BUILD')
-rw-r--r-- | tensorflow/python/kernel_tests/BUILD | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/tensorflow/python/kernel_tests/BUILD b/tensorflow/python/kernel_tests/BUILD index cac05c372a..896d466c25 100644 --- a/tensorflow/python/kernel_tests/BUILD +++ b/tensorflow/python/kernel_tests/BUILD @@ -1672,6 +1672,18 @@ cuda_py_test( ) cuda_py_test( + name = "compare_and_bitpack_op_test", + size = "small", + srcs = ["compare_and_bitpack_op_test.py"], + additional_deps = [ + "//third_party/py/numpy", + "//tensorflow/python:math_ops", + "//tensorflow/python:client_testlib", + "//tensorflow/python:framework_for_generated_wrappers", + ], +) + +cuda_py_test( name = "scalar_test", size = "small", srcs = ["scalar_test.py"], |