Searched defs:padded_shapes (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | padded_batch_dataset_op.cc | 52 bool parallel_copy, std::vector<PartialTensorShape> padded_shapes, in Dataset()
140 std::vector<Node*> padded_shapes; in AsGraphDefInternal() local
417 std::vector<PartialTensorShape> padded_shapes; in MakeDataset() local
|
| D | padded_batch_dataset_op_test.cc | 29 std::vector<Tensor> padded_shapes, in PaddedBatchDatasetParams()
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | padded_batch_test.py | 49 def testPaddedBatchDataset(self, count, padded_shapes, drop_remainder): argument
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | grouping.py | 131 padded_shapes=None, argument
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | dataset_ops.py | 1554 padded_shapes=None, argument
2643 padded_shapes=None, argument
4212 def __init__(self, input_dataset, batch_size, padded_shapes, padding_values, argument
|