| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | quantization_test.py | 74 def ShouldRunTest(self, run_params): argument
78 return (trt_test.IsQuantizationMode(run_params.precision_mode) and
79 not run_params.use_optimizer and not run_params.dynamic_engine)
81 def ExpectedEnginesToBuild(self, run_params): argument
83 if run_params.use_calibration:
98 def ShouldRunTest(self, run_params): argument
102 return (trt_test.IsQuantizationMode(run_params.precision_mode) and
103 not run_params.use_optimizer)
105 def ExpectedEnginesToBuild(self, run_params): argument
109 def ExpectedAbsoluteTolerance(self, run_params): argument
[all …]
|
| D | tf_trt_integration_test_base.py | 165 def GetConversionParams(self, run_params): argument
180 precision_mode=run_params.precision_mode,
182 is_dynamic_op=run_params.dynamic_engine,
186 use_calibration=run_params.use_calibration)
188 def ShouldRunTest(self, run_params): argument
192 return not (IsQuantizationMode(run_params.precision_mode) and
193 not run_params.use_calibration)
195 def ExpectedEnginesToBuild(self, run_params): argument
199 def ExpectedAbsoluteTolerance(self, run_params): argument
201 return 1.e-05 if run_params.precision_mode == "FP32" else 1.e-02
[all …]
|
| D | dynamic_input_shapes_test.py | 80 def GetConversionParams(self, run_params): argument
83 self).GetConversionParams(run_params)
90 def ExpectedEnginesToBuild(self, run_params): argument
93 def ShouldRunTest(self, run_params): argument
94 return (run_params.dynamic_engine and
95 not trt_test.IsQuantizationMode(run_params.precision_mode))
97 def ExpectedAbsoluteTolerance(self, run_params): argument
99 return 1.e-03 if run_params.precision_mode == "FP32" else 1.e-01
101 def ExpectedRelativeTolerance(self, run_params): argument
103 return 1.e-03 if run_params.precision_mode == "FP32" else 1.e-01
|
| D | reshape_transpose_test.py | 79 def ExpectedEnginesToBuild(self, run_params): argument
86 def ShouldRunTest(self, run_params): argument
88 return (not trt_test.IsQuantizationMode(run_params.precision_mode) and
89 not run_params.dynamic_engine)
136 def ExpectedEnginesToBuild(self, run_params): argument
145 def ShouldRunTest(self, run_params): argument
147 return (not trt_test.IsQuantizationMode(run_params.precision_mode) and
148 not run_params.dynamic_engine)
|
| D | base_test.py | 74 def ExpectedEnginesToBuild(self, run_params): argument
131 def ExpectedEnginesToBuild(self, run_params): argument
140 def GetConversionParams(self, run_params): argument
144 ).GetConversionParams(run_params)._replace(
183 def ExpectedEnginesToBuild(self, run_params): argument
190 def ShouldRunTest(self, run_params): argument
194 return ((run_params.precision_mode != "FP16") and
195 not (trt_test.IsQuantizationMode(run_params.precision_mode) and
196 not run_params.use_calibration))
236 def ExpectedEnginesToBuild(self, run_params): argument
[all …]
|
| D | int32_test.py | 59 def GetConversionParams(self, run_params): argument
62 self).GetConversionParams(run_params)
70 def ExpectedEnginesToBuild(self, run_params): argument
74 def ShouldRunTest(self, run_params): argument
78 return not trt_test.IsQuantizationMode(run_params.precision_mode)
|
| D | const_broadcast_test.py | 65 def ExpectedEnginesToBuild(self, run_params): argument
69 def ExpectedAbsoluteTolerance(self, run_params): argument
71 return 1.e-04 if run_params.precision_mode == 'FP32' else 1.e-02
73 def ExpectedRelativeTolerance(self, run_params): argument
75 return 1.e-04 if run_params.precision_mode == 'FP32' else 1.e-02
|
| D | memory_alignment_test.py | 69 def ExpectedEnginesToBuild(self, run_params): argument
73 def ExpectedAbsoluteTolerance(self, run_params): argument
75 return 1.e-06 if run_params.precision_mode == "FP32" else 1.e-02
77 def ExpectedRelativeTolerance(self, run_params): argument
|
| D | lru_cache_test.py | 68 def ExpectedEnginesToBuild(self, run_params): argument
72 def ShouldRunTest(self, run_params): argument
73 return (run_params.dynamic_engine and
74 not trt_test.IsQuantizationMode(run_params.precision_mode))
|
| D | conv2d_test.py | 111 def ExpectedEnginesToBuild(self, run_params): argument
136 def ExpectedEnginesToBuild(self, run_params): argument
186 def ExpectedEnginesToBuild(self, run_params): argument
227 def ExpectedEnginesToBuild(self, run_params): argument
|
| D | biasadd_matmul_test.py | 118 def GetConversionParams(self, run_params): argument
121 self).GetConversionParams(run_params)
129 def ExpectedEnginesToBuild(self, run_params): argument
|
| D | topk_test.py | 52 def ExpectedEnginesToBuild(self, run_params): argument
83 def ExpectedEnginesToBuild(self, run_params): argument
|
| D | identity_output_test.py | 68 def ExpectedEnginesToBuild(self, run_params): argument
|
| D | binary_tensor_weight_broadcast_test.py | 70 def ExpectedEnginesToBuild(self, run_params): argument
|
| D | neighboring_engine_test.py | 66 def ExpectedEnginesToBuild(self, run_params): argument
|
| D | vgg_block_test.py | 68 def ExpectedEnginesToBuild(self, run_params): argument
|
| D | vgg_block_nchw_test.py | 77 def ExpectedEnginesToBuild(self, run_params): argument
|
| D | multi_connection_neighbor_engine_test.py | 82 def ExpectedEnginesToBuild(self, run_params): argument
|
| D | rank_two_test.py | 70 def ExpectedEnginesToBuild(self, run_params): argument
|
| D | batch_matmul_test.py | 78 def ExpectedEnginesToBuild(self, run_params): argument
|
| D | concatenation_test.py | 80 def ExpectedEnginesToBuild(self, run_params): argument
|
| D | unary_test.py | 107 def ExpectedEnginesToBuild(self, run_params): argument
|