Home
last modified time | relevance | path

Searched refs:padding_amount (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/batching_util/
Dbatch_resource_base.cc325 const int padding_amount = padded_batch_size - batch.size(); in ConcatInputTensors() local
326 profiler::TraceMe trace_me([padded_batch_size, padding_amount]() { in ConcatInputTensors()
329 {"padding_amount", padding_amount}}); in ConcatInputTensors()
331 RecordPaddingSize(padding_amount, GetModelName(context), padded_batch_size, in ConcatInputTensors()
351 if (padding_amount > 0) { in ConcatInputTensors()
365 for (int i = 0; i < padding_amount; ++i) { in ConcatInputTensors()