Searched defs:padded_shapes (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | padded_batch_dataset_op.cc | 52 std::vector<PartialTensorShape> padded_shapes; in MakeDataset() local
104 std::vector<PartialTensorShape> padded_shapes, in Dataset()
173 std::vector<Node*> padded_shapes; in AsGraphDefInternal() local
|
| /external/tensorflow/tensorflow/contrib/data/python/ops/ |
| D | grouping.py | 100 padded_shapes=None, argument
|
| D | batching.py | 141 padded_shapes, argument
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | padded_batch_test.py | 51 def testPaddedBatchDataset(self, seq_lens, batch_size, padded_shapes, argument
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | grouping.py | 130 padded_shapes=None, argument
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu.py | 578 def _pad_all_input(inputs, padded_shapes): argument
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | dataset_ops.py | 873 padded_shapes, argument
1654 padded_shapes, argument
2907 def __init__(self, input_dataset, batch_size, padded_shapes, padding_values, argument
|