Searched refs:getBatchSize (Results 1 – 3 of 3) sorted by relevance
55 int getBatchSize() const override;
31 int TRTInt8Calibrator::getBatchSize() const { return batch_size_; } in getBatchSize() function in tensorflow::tensorrt::TRTInt8Calibrator
713 cres->calibrator_->getBatchSize(), workspace_size_bytes, shapes, in AllocateCalibrationResources()