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

Adding a RebatchDatasetOp that uses Grappler to modify the dataset graph to... 

Adding a RebatchDatasetOp that uses Grappler to modify the dataset graph to mutate the batch size of the Dataset.

Right now this basically does all the wiring through Grappler etc. and the logic is very minimal and basic. Subsequent CL's will make the logic expand to different scenarios.

PiperOrigin-RevId: 231983575
parent 90a8b12b
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