Searched refs:use_calibration (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | tf_trt_integration_test_base.py | 73 return IsQuantizationMode(params.precision_mode) and params.use_calibration
186 use_calibration=run_params.use_calibration)
193 not run_params.use_calibration)
230 conversion_params.use_calibration,
312 use_calibration=conversion_params.use_calibration,
324 assert conversion_params.use_calibration
454 run_params.use_calibration, node.name)
458 run_params.use_calibration and graph_state == GraphState.INFERENCE):
516 run_params.use_calibration):
571 for (use_optimizer, precision_mode, dynamic_engine, use_calibration) in opts:
[all …]
|
| D | quantization_test.py | 83 if run_params.use_calibration:
|
| D | quantization_mnist_test.py | 151 use_calibration=False,
|
| D | base_test.py | 196 not run_params.use_calibration))
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.h | 102 use_calibration(true) {} in EngineInfo()
119 bool use_calibration; member
154 TrtUniquePtrType<nvinfer1::ICudaEngine>* engine, bool use_calibration,
436 TrtPrecisionMode precision_mode, bool use_calibration);
467 bool use_calibration() const { return use_calibration_; } in use_calibration() function
|
| D | convert_graph.h | 72 bool use_calibration = true; member
90 bool use_calibration = true);
|
| D | convert_graph.cc | 106 std::vector<int> cached_engine_batches, bool use_calibration) { in ConvertGraphDefToTensorRT() argument
172 parameters["use_calibration"].set_b(use_calibration); in ConvertGraphDefToTensorRT()
503 (info.precision_mode == TrtPrecisionMode::INT8 && info.use_calibration); in CreateTRTNode()
520 info.use_calibration, in CreateTRTNode()
568 .Attr("use_calibration", info.use_calibration) in CreateTRTNode()
791 if (params.use_calibration && !params.use_function_backup) { in ConvertAfterShapes()
796 if (params.use_calibration) { in ConvertAfterShapes()
855 curr_engine.engine_type = ((params.is_dyn_op || params.use_calibration) in ConvertAfterShapes()
858 curr_engine.use_calibration = params.use_calibration; in ConvertAfterShapes()
|
| D | trt_optimization_pass.cc | 267 cp.use_calibration = use_calibration_; in Optimize()
|
| D | convert_nodes.cc | 994 TrtPrecisionMode precision_mode, bool use_calibration) in Converter() argument
997 use_calibration_(use_calibration) { in Converter()
1264 if (precision_mode() == TrtPrecisionMode::INT8 && !use_calibration()) { in PrepareTensorForShape()
1320 if (!use_calibration()) { in MaybeApplyQuantizationRanges()
1714 !params->converter->use_calibration()) { in BinaryTensorOpWeight()
3211 !params->converter->use_calibration()) { in ConvertRsqrt()
4060 TrtUniquePtrType<nvinfer1::ICudaEngine>* engine, bool use_calibration, in ConvertGraphDefToEngine() argument
4079 if (use_calibration) { in ConvertGraphDefToEngine()
4095 Converter converter(trt_network.get(), precision_mode, use_calibration); in ConvertGraphDefToEngine()
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| D | trt_convert.py | 471 use_calibration=True, argument
553 optimizer.parameter_map["use_calibration"].b = use_calibration
571 use_calibration=True, argument
686 precision_mode == TrtPrecisionMode.INT8 and use_calibration)
719 use_calibration=self._need_calibration,
872 use_calibration=False)
|