diff options
Diffstat (limited to 'tensorflow/contrib/signal/BUILD')
| -rw-r--r-- | tensorflow/contrib/signal/BUILD | 18 |
1 files changed, 17 insertions, 1 deletions
diff --git a/tensorflow/contrib/signal/BUILD b/tensorflow/contrib/signal/BUILD index 6ed2313dca..706de58a0a 100644 --- a/tensorflow/contrib/signal/BUILD +++ b/tensorflow/contrib/signal/BUILD @@ -22,8 +22,24 @@ py_library( ) cuda_py_tests( + name = "reconstruction_ops_test", + srcs = ["python/kernel_tests/reconstruction_ops_test.py"], + additional_deps = [ + ":signal_py", + "//third_party/py/numpy", + "//tensorflow/python:array_ops", + "//tensorflow/python:gradients", + "//tensorflow/python:math_ops", + "//tensorflow/python:client_testlib", + "//tensorflow/python:framework", + "//tensorflow/python:framework_for_generated_wrappers", + "//tensorflow/python:framework_test_lib", + "//tensorflow/python:platform_test", + ], +) + +cuda_py_tests( name = "shape_ops_test", - size = "small", srcs = ["python/kernel_tests/shape_ops_test.py"], additional_deps = [ ":signal_py", |