Searched refs:padding_source (Results 1 – 1 of 1) sorted by relevance
462 const Tensor& padding_source = batch.task(0).inputs.at(i); in ConcatInputTensors() local464 if (padding_source.shape().dim_size(0) == 0) { in ConcatInputTensors()468 i, " got shape ", padding_source.shape().DebugString(), ".)"); in ConcatInputTensors()470 if (padding_source.shape().dim_size(0) == 1) { in ConcatInputTensors()471 padding = padding_source; in ConcatInputTensors()473 padding = padding_source.Slice(0, 1); in ConcatInputTensors()