Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dbatch_kernels.cc355 const Tensor& padding_source = batch.task(0).inputs.at(i); in ConcatInputTensors() local
357 if (padding_source.shape().dim_size(0) == 1) { in ConcatInputTensors()
358 padding = padding_source; in ConcatInputTensors()
361 const DataType type = padding_source.dtype(); in ConcatInputTensors()
368 SplitCPU<type>(context, padding_source, slice_sizes, &slices); \ in ConcatInputTensors()