Searched refs:padded_batch (Results 1 – 24 of 24) sorted by relevance
53 lambda x: array_ops.fill([x], x)).padded_batch(91 [6, 5, 5, 5, 5]).map(lambda x: array_ops.fill([x], x)).padded_batch(100 [0, 0, 0, 0]).map(lambda x: array_ops.fill([x], x)).padded_batch(112 [1, 2, 3, 4]).map(fill).padded_batch(batch_size=2))125 [1, 2, 3, 4]).map(fill_tuple).padded_batch(batch_size=2))148 .padded_batch(188 padded_dataset = dataset.padded_batch(202 dynamic_padding_from_tensor_shapes = input_dataset.padded_batch(207 dynamic_padding_from_lists = input_dataset.padded_batch(209 dynamic_padding_from_lists_with_minus_one = input_dataset.padded_batch([all …]
101 ("PaddedBatch1", lambda: dataset_ops.Dataset.range(5).padded_batch(103 ("PaddedBatch2", lambda: dataset_ops.Dataset.range(5).padded_batch(106 lambda _: True).padded_batch(2, []), dataset_ops.UNKNOWN),108 lambda: dataset_ops.Dataset.range(5).repeat().padded_batch(2, []),
167 self._testUnaryInputs(lambda x: x.padded_batch(10, []))
41 lambda x: array_ops.fill([x], x)).padded_batch(58 fill_tuple).padded_batch(
50 window.padded_batch(150 window.padded_batch(280 return xs.padded_batch(300 window.padded_batch(302 window.padded_batch(
78 name: "padded_batch"
80 name: "padded_batch"
79 name: "padded_batch"
111 name: "padded_batch"
113 name: "padded_batch"
261 return grouped_dataset.padded_batch(
284 dataset = dataset.padded_batch(42, padded_shapes=test_case['shape'][1:])
165 ('PaddedBatch', lambda: dataset_ops.Dataset.range(5).padded_batch(2, [])),
1552 def padded_batch(self, member in DatasetV22640 @functools.wraps(DatasetV2.padded_batch)2641 def padded_batch(self, member in DatasetV12647 super(DatasetV1, self).padded_batch(batch_size, padded_shapes,
1861 `padded_batch()`. This functionality can be enabled through2999 …* Adding `drop_remainder` argument to `tf.data.Dataset.batch()` and `tf.data.Dataset.padded_batch(…