Searched refs:grouped_dataset (Results 1 – 1 of 1) sorted by relevance
2723 def batching_fn(bucket_id, grouped_dataset): argument2727 return grouped_dataset.batch(batch_size, drop_remainder=drop_remainder)2742 input_shapes = get_legacy_output_shapes(grouped_dataset)2745 return grouped_dataset.padded_batch(