Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/data/experimental/
Ddirected_interleave_dataset_op.cc260 PartialTensorShape output_tensorshape; in MostSpecificCompatibleShape() local
262 return output_tensorshape; in MostSpecificCompatibleShape()
267 output_tensorshape.Concatenate(dims1[d]); in MostSpecificCompatibleShape()
269 output_tensorshape.Concatenate(-1); in MostSpecificCompatibleShape()
271 return output_tensorshape; in MostSpecificCompatibleShape()
/external/tensorflow/tensorflow/core/kernels/data/
Dconcatenate_dataset_op.cc200 PartialTensorShape output_tensorshape({}); in MostSpecificCompatibleShape() local
205 output_tensorshape.AddDim(dims1[d]); in MostSpecificCompatibleShape()
207 output_tensorshape.AddDim(-1); in MostSpecificCompatibleShape()
209 return output_tensorshape; in MostSpecificCompatibleShape()