Commit 5b9e7a77 authored by Rohan Jain's avatar Rohan Jain Committed by TensorFlower Gardener
Browse files

Throwing an error if the batch size is not divisible by num_workers in... 

Throwing an error if the batch size is not divisible by num_workers in RebatchDataset. Also greatly simplifying the graph mutation logic in RebatchDataset now that we know that the constant value for batch_size is already computed.

PiperOrigin-RevId: 233609414
parent 3df272f7
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment