Searched defs:TRTInt8Calibrator (Results 1 – 2 of 2) sorted by relevance
39 struct TRTInt8Calibrator : public nvinfer1::IInt8EntropyCalibrator2 { struct75 const string& getCalibrationTableAsString() { return calibration_table_; } in getCalibrationTableAsString()81 mutex cond_mtx_;84 condition_variable cond_;87 bool done_;90 const std::unordered_map<string, std::pair<void*, size_t>> dev_buffers_;92 bool calib_running_;93 bool batch_is_set_;95 string engine_name_;96 string calibration_table_;
33 TRTInt8Calibrator::TRTInt8Calibrator( in TRTInt8Calibrator() function in tensorflow::tensorrt::TRTInt8Calibrator44 TRTInt8Calibrator::TRTInt8Calibrator(const string& calib_data) in TRTInt8Calibrator() function in tensorflow::tensorrt::TRTInt8Calibrator