diff options
Diffstat (limited to 'tensorflow/contrib/tensorrt/BUILD')
| -rw-r--r-- | tensorflow/contrib/tensorrt/BUILD | 20 |
1 files changed, 19 insertions, 1 deletions
diff --git a/tensorflow/contrib/tensorrt/BUILD b/tensorflow/contrib/tensorrt/BUILD index 9e8979bce4..5c16fcb760 100644 --- a/tensorflow/contrib/tensorrt/BUILD +++ b/tensorflow/contrib/tensorrt/BUILD @@ -455,7 +455,6 @@ cuda_py_tests( "test/multi_connection_neighbor_engine_test.py", "test/neighboring_engine_test.py", "test/rank_two_test.py", - "test/unary_test.py", "test/vgg_block_nchw_test.py", "test/vgg_block_test.py", ], @@ -471,6 +470,25 @@ cuda_py_tests( ], ) +cuda_py_tests( + name = "tf_trt_integration_test_no_oss", + srcs = [ + "test/unary_test.py", + ], + additional_deps = [ + ":tf_trt_integration_test_base", + "//tensorflow/python:client_testlib", + "//tensorflow/python:framework_test_lib", + ], + tags = [ + "no_cuda_on_cpu_tap", + "no_oss", # TODO(b/117274186): re-enable in OSS after crash fixed + "no_pip", # TODO(b/117274186): re-enable in OSS after crash fixed + "no_windows", + "nomac", + ], +) + cc_library( name = "utils", srcs = ["convert/utils.cc"], |