experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[TF:TRT] Enable concatenation_test and biasadd_matmul_test for TAP.

Browse Source 
PiperOrigin-RevId: 311604247
Change-Id: Ifca2be4bf2f40dc48f2beffb76bea94fe52101b4
This commit is contained in:
Bixia Zheng 2020-05-14 14:19:16 -07:00 committed by TensorFlower Gardener
parent 6db3caf99b
commit ec7ea83d9d
1 changed files with 2 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
tensorflow/python/compiler/tensorrt/BUILD
View File

@ -120,8 +120,10 @@ cuda_py_tests(
srcs = [ srcs = [
"test/base_test.py", "test/base_test.py",
"test/batch_matmul_test.py", "test/batch_matmul_test.py",
"test/biasadd_matmul_test.py",
"test/binary_tensor_weight_broadcast_test.py", "test/binary_tensor_weight_broadcast_test.py",
"test/combined_nms_test.py", "test/combined_nms_test.py",
"test/concatenation_test.py",
"test/const_broadcast_test.py", "test/const_broadcast_test.py",
"test/conv2d_test.py", "test/conv2d_test.py",
"test/dynamic_input_shapes_test.py", "test/dynamic_input_shapes_test.py",
@ -155,27 +157,6 @@ cuda_py_tests(
], ],
) )
cuda_py_tests(
name = "concatenation_test",
srcs = [
"test/biasadd_matmul_test.py",
"test/concatenation_test.py",
],
python_version = "PY3",
tags = [
"no_rocm",
"no_windows",
"nomac",
"notap", # b/140261407
],
xla_enable_strict_auto_jit = False,
deps = [
":tf_trt_integration_test_base",
"//tensorflow/python:client_testlib",
"//tensorflow/python:framework_test_lib",
],
)
cuda_py_test( cuda_py_test(
name = "quantization_mnist_test", name = "quantization_mnist_test",
srcs = ["test/quantization_mnist_test.py"], srcs = ["test/quantization_mnist_test.py"],