Searched refs:calibrator_ (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 199 std::unique_ptr<TRTInt8Calibrator> calibrator_; member in tensorflow::tensorrt::TRTEngineOp
384 calibrator_.reset(new TRTInt8Calibrator(calibration_data)); in TRTEngineOp()
526 OP_REQUIRES_ASYNC(ctx, calib_ctx->calibrator_->setBatch(input_data, *stream), in ExecuteCalibration()
993 use_calibration_, calibrator_.get(), cache_res); in GetEngine()
1045 cres->calibrator_.reset( in AllocateCalibrationResources()
1080 cres->calibrator_->getBatchSize(), this->workspace_size_, in AllocateCalibrationResources()
1082 cres->calibrator_.get(), &cres->engine_, /*use_calibration=*/true, in AllocateCalibrationResources()
1087 cres->calibrator_->setDone(); // Ignore further pushes in AllocateCalibrationResources()
1092 this->calibrator_ = std::move(cres->calibrator_); in AllocateCalibrationResources()
1097 cres->calibrator_->setDone(); // Ignore further pushes in AllocateCalibrationResources()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_lru_cache.cc | 36 TRTInt8Calibrator* raw_calibrator = calibrator_.get(); in TerminateCalibration()
|
| D | trt_lru_cache.h | 176 std::unique_ptr<TRTInt8Calibrator> calibrator_; variable
|