Home
last modified time | relevance | path

Searched refs:use_function_backup (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert_test.py177 use_function_backup=False): argument
192 use_function_backup=use_function_backup)
231 use_function_backup=need_calibration)
343 use_function_backup=False, argument
352 self.assertTrue(not use_function_backup and not expect_engine_is_run)
384 use_function_backup=False) # Disallow fallback.
410 def _TestStaticOp(self, use_function_backup): argument
422 use_function_backup=use_function_backup)
433 use_function_backup=use_function_backup,
440 use_function_backup=use_function_backup,
[all …]
Dtrt_convert.py472 use_function_backup=True): argument
554 optimizer.parameter_map["use_function_backup"].b = use_function_backup
572 use_function_backup=True): argument
687 self._use_function_backup = use_function_backup
691 if self._need_calibration and not use_function_backup:
720 use_function_backup=self._use_function_backup)
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.h76 bool use_function_backup = true; member
Dconvert_graph.cc560 params.use_function_backup in CreateTRTNode()
791 if (params.use_calibration && !params.use_function_backup) { in ConvertAfterShapes()
861 if (params.use_function_backup) { in ConvertAfterShapes()
Dtrt_optimization_pass.cc268 cp.use_function_backup = use_function_backup_; in Optimize()
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dtf_trt_integration_test_base.py231 use_function_backup=IsQuantizationWithCalibration(conversion_params))
313 use_function_backup=IsQuantizationWithCalibration(conversion_params))
Dquantization_mnist_test.py152 use_function_backup=False)