Commit 312e37ce authored by Anjali Sridhar's avatar Anjali Sridhar Committed by TensorFlower Gardener
Browse files

Add a require_static_shapes argument to DistributionStrategy class. This... 

Add a require_static_shapes argument to DistributionStrategy class. This allows us to identify if we need to set the drop_remainder option when creating Dataset objects.

PiperOrigin-RevId: 215633097
parent c1b3b0b9
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