| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.h | 98 use_calibration(true), in EngineInfo()
116 bool use_calibration; member
153 TrtUniquePtrType<nvinfer1::ICudaEngine>* engine, bool use_calibration,
364 TrtPrecisionMode precision_mode, bool use_calibration,
380 const bool use_calibration; member
393 TrtPrecisionMode precision_mode, bool use_calibration,
457 TrtPrecisionMode precision_mode, bool use_calibration,
497 bool use_calibration() const { return use_calibration_; } in use_calibration() function
610 Converter(TrtPrecisionMode precision_mode, bool use_calibration,
|
| D | convert_graph.cc | 84 return (params.is_dyn_op || params.use_calibration) in GetEngineType()
425 (info.precision_mode == TrtPrecisionMode::INT8 && info.use_calibration); in CreateTRTNode()
457 info.use_calibration, params.use_implicit_batch, in CreateTRTNode()
495 .Attr("use_calibration", info.use_calibration) in CreateTRTNode()
685 params.use_calibration) { in ConvertAfterShapes()
743 params.use_calibration, params.use_implicit_batch); in ConvertAfterShapes()
779 curr_engine.use_calibration = params.use_calibration; in ConvertAfterShapes()
|
| D | convert_graph.h | 47 bool use_calibration = true; member
|
| D | convert_graph_test.cc | 171 params.use_calibration = false; in RunConvertAfterShape()
|
| D | trt_optimization_pass.cc | 243 cp.use_calibration = use_calibration_; in Optimize()
|
| D | convert_nodes.cc | 1104 TrtPrecisionMode precision_mode, bool use_calibration, in OpConverterParams() argument
1112 use_calibration(use_calibration), in OpConverterParams()
1126 use_calibration(converter->use_calibration()), in OpConverterParams()
1138 TrtPrecisionMode precision_mode, bool use_calibration, in TrtNodeValidator() argument
1142 use_calibration_(use_calibration), in TrtNodeValidator()
1243 TrtPrecisionMode precision_mode, bool use_calibration, in Create() argument
1247 new Converter(precision_mode, use_calibration, trt_logger, in Create()
1253 Converter::Converter(TrtPrecisionMode precision_mode, bool use_calibration, in Converter() argument
1258 use_calibration_(use_calibration), in Converter()
1750 !converter->use_calibration()) { in PrepareTensorForShape()
[all …]
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | tf_trt_integration_test_base.py | 104 return IsQuantizationMode(params.precision_mode) and params.use_calibration
284 use_calibration=run_params.use_calibration)
303 return (run_params.use_calibration or not IsQuantizationMode(
471 use_calibration=conversion_params.use_calibration)
483 assert conversion_params.use_calibration
841 run_params.use_calibration, node.name)
1053 use_calibration, no_calibration = True, False
1064 opts.append((INT8, convert_offline, dynamic_engine, use_calibration))
1111 for (precision_mode, convert_online, dynamic_engine, use_calibration) in opts:
1114 calibration_type = "UseCalibration" if use_calibration else "NoCalibration"
[all …]
|
| D | trt_mode_test.py | 124 not run_params.use_calibration), "test v2, >=TRT6 and non-calibration"
161 not run_params.use_calibration), "test v2 >=TRT6 and non-calibration"
|
| D | base_test.py | 165 not run_params.use_calibration)), "test FP32 and non-calibration"
|
| D | quantization_mnist_test.py | 156 use_calibration=False)
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| D | trt_convert.py | 160 use_calibration=True, argument
165 use_calibration, allow_build_at_runtime)
297 "use_calibration"].b = conversion_params.use_calibration
383 use_calibration=True): argument
452 precision_mode == TrtPrecisionMode.INT8 and use_calibration)
476 use_calibration=use_calibration,
961 conversion_params.use_calibration)
1314 use_calibration=False)
|
| D | trt_convert_windows.py | 88 use_calibration=True): argument
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.experimental.tensorrt.-conversion-params.pbtxt | 27 name: "use_calibration"
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/model_tests/ |
| D | model_handler.py | 463 use_calibration=trt_convert_params.use_calibration,
477 self.trt_convert_params.use_calibration):
526 self.trt_convert_params.use_calibration):
|
| D | run_models.py | 157 use_calibration=(precision_mode == trt.TrtPrecisionMode.INT8))
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 146 bool use_calibration, TRTInt8Calibrator* calibrator,
829 bool use_calibration, TRTInt8Calibrator* calibrator, in BuildEngine() argument
846 calibrator, &engine, use_calibration, use_implicit_batch_, nullptr, in BuildEngine()
|