Searched refs:grouped_dataset (Results 1 – 1 of 1) sorted by relevance
239 def batching_fn(bucket_id, grouped_dataset): argument243 return grouped_dataset.batch(batch_size, drop_remainder=drop_remainder)258 input_shapes = dataset_ops.get_legacy_output_shapes(grouped_dataset)261 return grouped_dataset.padded_batch(