Searched refs:setBatch (Results 1 – 3 of 3) sorted by relevance
61 bool setBatch(const std::unordered_map<string, void*>& data,
50 bool TRTInt8Calibrator::setBatch(const std::unordered_map<string, void*>& data, in setBatch() function in tensorflow::tensorrt::TRTInt8Calibrator
526 OP_REQUIRES_ASYNC(ctx, calib_ctx->calibrator_->setBatch(input_data, *stream), in ExecuteCalibration()