| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | padded_batch_test.py | 47 padded_shapes=[[None], [25]],
49 def testPaddedBatchDataset(self, count, padded_shapes, drop_remainder): argument
56 padded_shapes=padded_shapes)
62 padded_len = padded_shapes[0]
73 padded_len = padded_shapes[0]
92 batch_size=4, padded_shapes=[5]))
101 batch_size=4, padded_shapes=[-1]))
149 4, padded_shapes=([-1], [-1], {'structure': [-1]}),
189 2, padded_shapes=([None], [None]), padding_values=('', 0))
204 padded_shapes=(tensor_shape.TensorShape([None]),
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | padded_batch_dataset_op.cc | 52 bool parallel_copy, std::vector<PartialTensorShape> padded_shapes, in Dataset() argument
59 padded_shapes_(std::move(padded_shapes)), in Dataset()
140 std::vector<Node*> padded_shapes; in AsGraphDefInternal() local
141 padded_shapes.reserve(padded_shapes_.size()); in AsGraphDefInternal()
149 padded_shapes.emplace_back(node); in AsGraphDefInternal()
174 {{2, padded_shapes}, {3, padding_values}}, in AsGraphDefInternal()
417 std::vector<PartialTensorShape> padded_shapes; in MakeDataset() local
418 padded_shapes.reserve(padded_shape_tensors.size()); in MakeDataset()
432 padded_shapes.push_back(std::move(padded_shape)); in MakeDataset()
457 std::move(padded_shapes), std::move(padding_values), in MakeDataset()
|
| D | padded_batch_dataset_op_test.cc | 29 std::vector<Tensor> padded_shapes, in PaddedBatchDatasetParams() argument
38 padded_shapes_(std::move(padded_shapes)), in PaddedBatchDatasetParams()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | PaddedBatchDataset.pbtxt | 12 name: "padded_shapes"
55 name: "padded_shapes"
|
| D | PaddedBatchDatasetV2.pbtxt | 12 name: "padded_shapes"
58 name: "padded_shapes"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | PaddedBatchDataset.pbtxt | 12 name: "padded_shapes"
55 name: "padded_shapes"
|
| D | PaddedBatchDatasetV2.pbtxt | 12 name: "padded_shapes"
58 name: "padded_shapes"
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | padded_batch_dataset_serialization_test.py | 42 4, padded_shapes=[-1])
60 padded_shapes=(padded_shape, padded_shape),
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_PaddedBatchDataset.pbtxt | 12 name: "padded_shapes"
|
| D | api_def_PaddedBatchDatasetV2.pbtxt | 19 name: "padded_shapes"
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | group_by_window_test.py | 282 padded_shapes=(tensor_shape.TensorShape([]),
301 4, padded_shapes=tensor_shape.TensorShape([None])),
303 4, padded_shapes=ops.convert_to_tensor([(key + 1) * 10])),
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | grouping.py | 131 padded_shapes=None, argument
259 shapes = make_padded_shapes(padded_shapes or input_shapes,
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | dataset_ops.py | 1554 padded_shapes=None, argument
1667 if padded_shapes is None:
1668 padded_shapes = get_legacy_output_shapes(self)
1671 if not all(nest.flatten(padded_shapes)):
1675 return PaddedBatchDataset(self, batch_size, padded_shapes, padding_values,
2643 padded_shapes=None, argument
2647 super(DatasetV1, self).padded_batch(batch_size, padded_shapes,
4212 def __init__(self, input_dataset, batch_size, padded_shapes, padding_values, argument
4229 flat_padded_shapes = nest.flatten_up_to(input_shapes, padded_shapes)
4271 padded_shapes=[
[all …]
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.data.-dataset.pbtxt | 79 …argspec: "args=[\'self\', \'batch_size\', \'padded_shapes\', \'padding_values\', \'drop_remainder\…
|
| D | tensorflow.data.-t-f-record-dataset.pbtxt | 80 …argspec: "args=[\'self\', \'batch_size\', \'padded_shapes\', \'padding_values\', \'drop_remainder\…
|
| D | tensorflow.data.-text-line-dataset.pbtxt | 81 …argspec: "args=[\'self\', \'batch_size\', \'padded_shapes\', \'padding_values\', \'drop_remainder\…
|
| D | tensorflow.data.experimental.-random-dataset.pbtxt | 81 …argspec: "args=[\'self\', \'batch_size\', \'padded_shapes\', \'padding_values\', \'drop_remainder\…
|
| D | tensorflow.data.experimental.-sql-dataset.pbtxt | 81 …argspec: "args=[\'self\', \'batch_size\', \'padded_shapes\', \'padding_values\', \'drop_remainder\…
|
| D | tensorflow.data.experimental.-csv-dataset.pbtxt | 81 …argspec: "args=[\'self\', \'batch_size\', \'padded_shapes\', \'padding_values\', \'drop_remainder\…
|
| D | tensorflow.data.-fixed-length-record-dataset.pbtxt | 81 …argspec: "args=[\'self\', \'batch_size\', \'padded_shapes\', \'padding_values\', \'drop_remainder\…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.data.-dataset.pbtxt | 112 …argspec: "args=[\'self\', \'batch_size\', \'padded_shapes\', \'padding_values\', \'drop_remainder\…
|
| D | tensorflow.data.experimental.-random-dataset.pbtxt | 114 …argspec: "args=[\'self\', \'batch_size\', \'padded_shapes\', \'padding_values\', \'drop_remainder\…
|
| D | tensorflow.data.experimental.-sql-dataset.pbtxt | 114 …argspec: "args=[\'self\', \'batch_size\', \'padded_shapes\', \'padding_values\', \'drop_remainder\…
|
| D | tensorflow.data.-t-f-record-dataset.pbtxt | 114 …argspec: "args=[\'self\', \'batch_size\', \'padded_shapes\', \'padding_values\', \'drop_remainder\…
|
| D | tensorflow.data.-text-line-dataset.pbtxt | 114 …argspec: "args=[\'self\', \'batch_size\', \'padded_shapes\', \'padding_values\', \'drop_remainder\…
|