Commit b8fa2000 authored by Priya Gupta's avatar Priya Gupta Committed by TensorFlower Gardener
Browse files

Distribution Strategy: Fix batch splitting in case of map_and_batch. Also add unittests. 

PiperOrigin-RevId: 225852554
parent f2a08fbe
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