Searched refs:conversion_params (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| D | trt_convert.py | 173 def _check_conversion_params(conversion_params, is_v2=False): argument
185 if conversion_params.precision_mode not in supported_precision_modes:
188 "It should be one of {}").format(conversion_params.precision_mode,
232 def _get_tensorrt_rewriter_config(conversion_params, argument
255 _check_conversion_params(conversion_params, is_v2=is_v2)
289 "minimum_segment_size"].i = conversion_params.minimum_segment_size
291 conversion_params.max_workspace_size_bytes)
293 conversion_params.precision_mode)
295 "maximum_cached_engines"].i = conversion_params.maximum_cached_engines
297 "use_calibration"].b = conversion_params.use_calibration
[all …]
|
| D | trt_convert_windows.py | 104 conversion_params=None): argument
|
| D | trt_convert_test.py | 353 conversion_params = trt_convert.DEFAULT_TRT_CONVERSION_PARAMS._replace(
357 conversion_params,
387 conversion_params=trt_convert.DEFAULT_TRT_CONVERSION_PARAMS._replace(
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | tf_trt_integration_test_base.py | 276 conversion_params = trt_convert.TrtConversionParams(
285 return conversion_params
334 conversion_params = self.GetConversionParams(run_params)
339 conversion_params,
451 def _CreateConverter(self, run_params, saved_model_dir, conversion_params): argument
456 conversion_params=conversion_params)
466 max_workspace_size_bytes=conversion_params.max_workspace_size_bytes,
467 precision_mode=conversion_params.precision_mode,
468 minimum_segment_size=conversion_params.minimum_segment_size,
470 maximum_cached_engines=conversion_params.maximum_cached_engines,
[all …]
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.experimental.tensorrt.-converter.pbtxt | 7 … \'input_saved_model_tags\', \'input_saved_model_signature_key\', \'conversion_params\'], varargs=…
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/model_tests/ |
| D | model_handler.py | 512 conversion_params=trt_convert_params)
|
| /external/tensorflow/tensorflow/python/tools/ |
| D | saved_model_cli.py | 788 conversion_params=params)
|