Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/data/
Dpadded_batch_dataset_op.cc415 OpInputList padded_shape_tensors; in MakeDataset() local
416 OP_REQUIRES_OK(ctx, ctx->input_list(kPaddedShapes, &padded_shape_tensors)); in MakeDataset()
418 padded_shapes.reserve(padded_shape_tensors.size()); in MakeDataset()
419 OP_REQUIRES(ctx, padded_shape_tensors.size() == input->output_shapes().size(), in MakeDataset()
421 padded_shape_tensors.size(), in MakeDataset()
425 for (const Tensor& padded_shape_t : padded_shape_tensors) { in MakeDataset()