Home
last modified time | relevance | path

Searched refs:use_calibration_ (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dtrt_optimization_pass.h43 use_calibration_(true), in name_()
72 bool use_calibration_; variable
Dtrt_optimization_pass.cc73 use_calibration_ = params.at("use_calibration").b(); in Init()
232 if (use_calibration_ && precision_mode_ != TrtPrecisionMode::INT8) { in Optimize()
236 use_calibration_ = false; in Optimize()
267 cp.use_calibration = use_calibration_; in Optimize()
Dconvert_nodes.h467 bool use_calibration() const { return use_calibration_; } in use_calibration()
573 const bool use_calibration_; variable
Dconvert_nodes.cc997 use_calibration_(use_calibration) { in Converter()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc140 bool use_calibration_; member in tensorflow::tensorrt::TRTEngineOp
219 context->GetAttr("use_calibration", &use_calibration_)); in TRTEngineOp()
221 (use_calibration_ && precision_mode_ == TrtPrecisionMode::INT8 && in TRTEngineOp()
626 use_calibration_, &convert_successfully); in GetEngine()