Home
last modified time | relevance | path

Searched defs:padded_shapes (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dpadded_batch_dataset_op.cc53 bool parallel_copy, std::vector<PartialTensorShape> padded_shapes, in Dataset()
142 std::vector<Node*> padded_shapes; in AsGraphDefInternal() local
440 std::vector<PartialTensorShape> padded_shapes; in MakeDataset() local
Dpadded_batch_dataset_op_test.cc29 std::vector<Tensor> padded_shapes, in PaddedBatchDatasetParams()
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dgrouping.py116 padded_shapes=None, argument
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dpadded_batch_test.py46 def testPaddedBatchDataset(self, count, padded_shapes, drop_remainder): argument
/external/tensorflow/tensorflow/python/data/ops/
Ddataset_ops.py1903 padded_shapes=None, argument
2974 padded_shapes=None, argument
3990 padded_shapes=None, argument
5315 padded_shapes, argument