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()