Searched refs:calibrator_ (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_resources.cc | 38 oss << " Calibrator = " << hex << calibrator_.get() << dec << endl in DebugString()
48 calibrator_->waitAndSetDone(); in SerializeToString()
50 *serialized = calibrator_->getCalibrationTableAsString(); in SerializeToString()
|
| D | trt_resources.h | 59 std::unique_ptr<TRTInt8Calibrator> calibrator_; variable
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 136 std::unique_ptr<TRTInt8Calibrator> calibrator_; member in tensorflow::tensorrt::TRTEngineOp
224 calibrator_.reset(new TRTInt8Calibrator(calibration_data)); in TRTEngineOp()
328 calib_res->calibrator_->setBatch(input_data, *stream); in ExecuteCalibration()
625 partial_shapes, &logger, allocator, calibrator_.get(), &engine, in GetEngine()
682 cres->calibrator_.reset( in AllocateCalibrationResources()
713 cres->calibrator_->getBatchSize(), workspace_size_bytes, shapes, in AllocateCalibrationResources()
714 &cres->logger_, cres->allocator_.get(), cres->calibrator_.get(), in AllocateCalibrationResources()
720 cres->calibrator_->setDone(); // Ignore further pushes in AllocateCalibrationResources()
|