Searched refs:TRTInt8Calibrator (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_int8_calibrator.cc | 31 int TRTInt8Calibrator::getBatchSize() const { return batch_size_; } in getBatchSize()
33 TRTInt8Calibrator::TRTInt8Calibrator( in TRTInt8Calibrator() function in tensorflow::tensorrt::TRTInt8Calibrator
44 TRTInt8Calibrator::TRTInt8Calibrator(const string& calib_data) in TRTInt8Calibrator() function in tensorflow::tensorrt::TRTInt8Calibrator
51 bool TRTInt8Calibrator::setBatch(const std::unordered_map<string, void*>& data, in setBatch()
88 bool TRTInt8Calibrator::getBatch(void** bindings, const char** names, in getBatch()
113 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 | 39 struct TRTInt8Calibrator : public nvinfer1::IInt8EntropyCalibrator2 { struct
41 struct TRTInt8Calibrator : public nvinfer1::IInt8EntropyCalibrator {
45 TRTInt8Calibrator(
51 TRTInt8Calibrator(const string& calibration_data);
53 ~TRTInt8Calibrator();
|
| D | trt_resources.h | 59 std::unique_ptr<TRTInt8Calibrator> calibrator_;
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 136 std::unique_ptr<TRTInt8Calibrator> calibrator_;
224 calibrator_.reset(new TRTInt8Calibrator(calibration_data)); in TRTEngineOp()
683 new TRTInt8Calibrator(cres->device_buffers_, batch_size, name())); in AllocateCalibrationResources()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.h | 153 nvinfer1::IGpuAllocator* allocator, TRTInt8Calibrator* calibrator,
|
| D | convert_nodes.cc | 4059 nvinfer1::IGpuAllocator* allocator, TRTInt8Calibrator* calibrator, in ConvertGraphDefToEngine()
|