Commit 1885344c authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

Explictly raise an exception if sharding fails because the batch size is unknown. 

Right now, TPUEstimator will raise an error message complaining that sharding
failed because (None % integer value) is not a well-defined operation. I hope
the new error message will make it easier for people to figure out what's going
on.

PiperOrigin-RevId: 172013663
parent 8c57896d
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