Searched refs:padding_source (Results 1 – 1 of 1) sorted by relevance
423 const Tensor& padding_source = batch.task(0).inputs.at(i); in ConcatInputTensors() local425 if (padding_source.shape().dim_size(0) == 0) { in ConcatInputTensors()429 i, " got shape ", padding_source.shape().DebugString(), ".)"); in ConcatInputTensors()431 if (padding_source.shape().dim_size(0) == 1) { in ConcatInputTensors()432 padding = padding_source; in ConcatInputTensors()434 padding = padding_source.Slice(0, 1); in ConcatInputTensors()