Searched refs:use_calibration_ (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | trt_optimization_pass.h | 43 use_calibration_(true), in name_()
76 bool use_calibration_; variable
|
| D | trt_optimization_pass.cc | 68 use_calibration_ = params.at("use_calibration").b(); in Init()
207 if (use_calibration_ && precision_mode_ != TrtPrecisionMode::INT8) { in Optimize()
211 use_calibration_ = false; in Optimize()
243 cp.use_calibration = use_calibration_; in Optimize()
|
| D | convert_nodes.h | 432 const bool use_calibration_; variable
497 bool use_calibration() const { return use_calibration_; } in use_calibration()
667 const bool use_calibration_; variable
|
| D | convert_nodes.cc | 1142 use_calibration_(use_calibration), in TrtNodeValidator()
1223 &weight_store_, precision_mode_, use_calibration_, in IsTensorRTCandidate()
1234 precision_mode_, use_calibration_, in ConvertConstToWeights()
1258 use_calibration_(use_calibration), in Converter()
1490 if (use_calibration_) { in BuildCudaEngine()
1507 "Precision:", precision_mode_str, ", ", "Calibration:", use_calibration_, in BuildCudaEngine()
1526 if (use_calibration_) { in BuildCudaEngine()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 203 bool use_calibration_; member in tensorflow::tensorrt::TRTEngineOp
351 context->GetAttr("use_calibration", &use_calibration_)); in TRTEngineOp()
381 (use_calibration_ && precision_mode_ == TrtPrecisionMode::INT8 && in TRTEngineOp()
993 use_calibration_, calibrator_.get(), cache_res); in GetEngine()
|