Searched refs:grouped_dataset (Results 1 – 1 of 1) sorted by relevance
3081 def batching_fn(bucket_id, grouped_dataset): argument3085 return grouped_dataset.batch(3101 input_shapes = get_legacy_output_shapes(grouped_dataset)3104 return grouped_dataset.padded_batch(