Searched refs:getBatchSize (Results 1 – 3 of 3) sorted by relevance
54 int getBatchSize() const override;
30 int TRTInt8Calibrator::getBatchSize() const { return batch_size_; } in getBatchSize() function in tensorflow::tensorrt::TRTInt8Calibrator
1080 cres->calibrator_->getBatchSize(), this->workspace_size_, in AllocateCalibrationResources()