Searched refs:padded_batch (Results 1 – 25 of 29) sorted by relevance
12
64 lambda x: array_ops.fill([x], x)).padded_batch(100 [6, 5, 5, 5, 5]).map(lambda x: array_ops.fill([x], x)).padded_batch(108 [0, 0, 0, 0]).map(lambda x: array_ops.fill([x], x)).padded_batch(121 .padded_batch(154 padded_dataset = dataset.padded_batch(169 dynamic_padding_from_tensor_shapes = input_dataset.padded_batch(174 dynamic_padding_from_lists = input_dataset.padded_batch(176 dynamic_padding_from_lists_with_minus_one = input_dataset.padded_batch(178 dynamic_padding_from_tensors = input_dataset.padded_batch(200 _ = dataset_ops.Dataset.range(10).map(_map_fn).padded_batch(10)[all …]
123 lambda x: x.padded_batch(10, []),
92 ("PaddedBatch1", lambda: dataset_ops.Dataset.range(5).padded_batch(94 ("PaddedBatch2", lambda: dataset_ops.Dataset.range(5).padded_batch(97 lambda _: True).padded_batch(2, []), cardinality.UNKNOWN),99 lambda: dataset_ops.Dataset.range(5).repeat().padded_batch(2, []),
50 window.padded_batch(147 window.padded_batch(271 return xs.padded_batch(290 window.padded_batch(292 window.padded_batch(
36 lambda x: array_ops.fill([x], x)).padded_batch(54 fill_tuple).padded_batch(
168 return dataset.padded_batch(
126 ('PaddedBatch', lambda: dataset_ops.Dataset.range(5).padded_batch(2, []),170 ('PaddedBatch', lambda: dataset_ops.Dataset.range(5).padded_batch(2, [])),
62 name: "padded_batch"
64 name: "padded_batch"
63 name: "padded_batch"
95 name: "padded_batch"
97 name: "padded_batch"
131 buffer_size=10000).padded_batch(
235 return grouped_dataset.padded_batch(
871 def padded_batch(self, member in DatasetV21651 @functools.wraps(DatasetV2.padded_batch)1652 def padded_batch(self, member in DatasetV11657 return DatasetV1Adapter(super(DatasetV1, self).padded_batch(
283 dataset = dataset.padded_batch(42, padded_shapes=test_case['shape'][1:])