Searched refs:grouped_dataset (Results 1 – 1 of 1) sorted by relevance
213 def batching_fn(bucket_id, grouped_dataset): argument217 return grouped_dataset.batch(batch_size, drop_remainder=drop_remainder)232 input_shapes = dataset_ops.get_legacy_output_shapes(grouped_dataset)235 return grouped_dataset.padded_batch(