Searched refs:padded_batch_size (Results 1 – 3 of 3) sorted by relevance
34 self.padded_batch_size = 059 missing_count = self.padded_batch_size - original_batch_size76 missing_count = (self.padded_batch_size -
455 padding_handler.padded_batch_size = batch_size
324 const int padded_batch_size = RoundToLowestAllowedBatchSize(batch.size()); in ConcatInputTensors() local325 const int padding_amount = padded_batch_size - batch.size(); in ConcatInputTensors()326 profiler::TraceMe trace_me([padded_batch_size, padding_amount]() { in ConcatInputTensors()328 "ConcatInputTensors", {{"batch_size_after_padding", padded_batch_size}, in ConcatInputTensors()331 RecordPaddingSize(padding_amount, GetModelName(context), padded_batch_size, in ConcatInputTensors()333 RecordProcessedBatchSize(padded_batch_size, GetModelName(context), in ConcatInputTensors()