From bcb89187a398000d80c7c0b0ac5152e76edd2666 Mon Sep 17 00:00:00 2001 From: Guangda Lai Date: Tue, 21 May 2019 09:26:41 -0700 Subject: [PATCH] Remove TRT 4.0 restrictions on int32 test. PiperOrigin-RevId: 249264857 --- tensorflow/python/compiler/tensorrt/test/int32_test.py | 6 ------ 1 file changed, 6 deletions(-) diff --git a/tensorflow/python/compiler/tensorrt/test/int32_test.py b/tensorflow/python/compiler/tensorrt/test/int32_test.py index 41a5a27addc..9d1d97b1d80 100644 --- a/tensorflow/python/compiler/tensorrt/test/int32_test.py +++ b/tensorflow/python/compiler/tensorrt/test/int32_test.py @@ -71,12 +71,6 @@ class ExcludeUnsupportedInt32Test(trt_test.TfTrtIntegrationTestBase): """Return the expected engines to build.""" return [] - def ShouldRunTest(self, run_params): - """Whether to run the test.""" - # TODO(aaroey): Trt 4.0 forbids conversion for tensors with rank <3 in int8 - # mode, which is a bug. Re-enable this when trt library is fixed. - return not trt_test.IsQuantizationMode(run_params.precision_mode) - if __name__ == '__main__': test.main()