Searched refs:RoundToLowestAllowedBatchSize (Results 1 – 1 of 1) sorted by relevance
316 int RoundToLowestAllowedBatchSize(int batch_size) const { in RoundToLowestAllowedBatchSize() function in tensorflow::BatchResource336 const int padded_batch_size = RoundToLowestAllowedBatchSize(batch.size()); in ConcatInputTensors()421 RoundToLowestAllowedBatchSize(batch->size()) - batch->size(); in SplitOutputTensors()