File tree Expand file tree Collapse file tree 1 file changed +8
-0
lines changed
tensorflow/python/compiler/tensorrt/test Expand file tree Collapse file tree 1 file changed +8
-0
lines changed Original file line number Diff line number Diff line change @@ -164,6 +164,14 @@ def ExpectedEnginesToBuild(self, run_params):
164
164
"""Return the expected engines to build."""
165
165
return ["TRTEngineOp_000" ]
166
166
167
+ def ExpectedAbsoluteTolerance (self , run_params ):
168
+ """The absolute tolerance to compare floating point results."""
169
+ return 5.e-01 if run_params .precision_mode == "INT8" else 1.e-02
170
+
171
+ def ExpectedRelativeTolerance (self , run_params ):
172
+ """The relative tolerance to compare floating point results."""
173
+ return 5.e-00 if run_params .precision_mode == "INT8" else 1.e-02
174
+
167
175
168
176
class Conv2DTranposeTest (trt_test .TfTrtIntegrationTestBase ):
169
177
"""Testing conversion of conv2d_transpose (AKA Conv2DBackpropInput)"""
You can’t perform that action at this time.
0 commit comments