Searched refs:setBatch (Results 1 – 3 of 3) sorted by relevance
60 bool setBatch(const std::unordered_map<string, void*>& data,
51 bool TRTInt8Calibrator::setBatch(const std::unordered_map<string, void*>& data, in setBatch() function in tensorflow::tensorrt::TRTInt8Calibrator
328 calib_res->calibrator_->setBatch(input_data, *stream); in ExecuteCalibration()