Searched refs:padding_source (Results 1 – 1 of 1) sorted by relevance
352 const Tensor& padding_source = batch.task(0).inputs.at(i); in ConcatInputTensors() local354 if (padding_source.shape().dim_size(0) == 0) { in ConcatInputTensors()358 i, " got shape ", padding_source.shape().DebugString(), ".)"); in ConcatInputTensors()360 if (padding_source.shape().dim_size(0) == 1) { in ConcatInputTensors()361 padding = padding_source; in ConcatInputTensors()363 padding = padding_source.Slice(0, 1); in ConcatInputTensors()