def ShouldRunTest(self, run_params):
# Although test passes with all configurations but only
# execute INT8 with use_calibration=True because
# that is the purpose of the test.
return trt_test.IsQuantizationWithCalibration(run_params)
def ExpectedRelativeTolerance(self, run_params):
"""The relative tolerance to compare floating point results."""
if trt_test.IsQuantizationWithCalibration(run_params):
return 4e-02
return super(Conv2DNCHWTest,
self).ExpectedRelativeTolerance(run_params)
