Searched refs:TRTInt8Calibrator (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_int8_calibrator.cc | 30 int TRTInt8Calibrator::getBatchSize() const { return batch_size_; } in getBatchSize()
32 TRTInt8Calibrator::TRTInt8Calibrator( in TRTInt8Calibrator() function in tensorflow::tensorrt::TRTInt8Calibrator
43 TRTInt8Calibrator::TRTInt8Calibrator(const string& calib_data) in TRTInt8Calibrator() function in tensorflow::tensorrt::TRTInt8Calibrator
50 bool TRTInt8Calibrator::setBatch(const std::unordered_map<string, void*>& data, in setBatch()
87 bool TRTInt8Calibrator::getBatch(void** bindings, const char** names, in getBatch()
112 void TRTInt8Calibrator::waitAndSetDone() { in waitAndSetDone()
124 const void* TRTInt8Calibrator::readCalibrationCache(std::size_t& length) { in readCalibrationCache()
130 void TRTInt8Calibrator::setDone() { in setDone()
136 void TRTInt8Calibrator::writeCalibrationCache(const void* ptr, in writeCalibrationCache()
142 TRTInt8Calibrator::~TRTInt8Calibrator() { in ~TRTInt8Calibrator()
|
| D | trt_int8_calibrator.h | 38 struct TRTInt8Calibrator : public nvinfer1::IInt8EntropyCalibrator2 { struct
40 struct TRTInt8Calibrator : public nvinfer1::IInt8EntropyCalibrator {
44 TRTInt8Calibrator(
50 TRTInt8Calibrator(const string& calibration_data);
52 ~TRTInt8Calibrator();
|
| D | trt_lru_cache.cc | 36 TRTInt8Calibrator* raw_calibrator = calibrator_.get(); in TerminateCalibration()
|
| D | trt_lru_cache.h | 176 std::unique_ptr<TRTInt8Calibrator> calibrator_;
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 146 bool use_calibration, TRTInt8Calibrator* calibrator,
199 std::unique_ptr<TRTInt8Calibrator> calibrator_;
384 calibrator_.reset(new TRTInt8Calibrator(calibration_data)); in TRTEngineOp()
829 bool use_calibration, TRTInt8Calibrator* calibrator, in BuildEngine()
1046 new TRTInt8Calibrator(cres->device_buffers_, batch_size, name())); in AllocateCalibrationResources()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.h | 152 TRTInt8Calibrator* calibrator,
482 TRTInt8Calibrator* calibrator,
|
| D | convert_nodes.cc | 1452 TRTInt8Calibrator* calibrator, TrtShapeOptimizationProfile* profiles) { in BuildCudaEngine()
6324 TRTInt8Calibrator* calibrator, in ConvertGraphDefToEngine()
|