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

Remove TRT 4.0 restrictions on int32 test.

Browse Source 
PiperOrigin-RevId: 249264857
This commit is contained in:
Guangda Lai 2019-05-21 09:26:41 -07:00 committed by TensorFlower Gardener
parent 9d46f5599f
commit bcb89187a3
1 changed files with 0 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/python/compiler/tensorrt/test/int32_test.py
View File

@ -71,12 +71,6 @@ class ExcludeUnsupportedInt32Test(trt_test.TfTrtIntegrationTestBase):
"""Return the expected engines to build.""" """Return the expected engines to build."""
return [] 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__': if __name__ == '__main__':
test.main() test.main()