Home
last modified time | relevance | path

Searched refs:calibration_table_ (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/
Dtrt_int8_calibrator.cc48 calibration_table_(calib_data) {} in TRTInt8Calibrator()
125 if (calibration_table_.empty()) return nullptr; in readCalibrationCache()
126 length = calibration_table_.size(); in readCalibrationCache()
127 return calibration_table_.data(); in readCalibrationCache()
138 calibration_table_ = string(static_cast<const char*>(ptr), length); in writeCalibrationCache()
Dtrt_lru_cache.cc34 if (terminated_) return calibration_table_; in TerminateCalibration()
48 calibration_table_ = raw_calibrator->getCalibrationTableAsString(); in TerminateCalibration()
49 return calibration_table_; in TerminateCalibration()
Dtrt_int8_calibrator.h76 const string& getCalibrationTableAsString() { return calibration_table_; } in getCalibrationTableAsString()
97 string calibration_table_; member
Dtrt_lru_cache.h185 std::string calibration_table_ TF_GUARDED_BY(mu_);