Commit 785b7bee authored by Sourabh Bajaj's avatar Sourabh Bajaj Committed by TensorFlower Gardener
Browse files

Fix to remove the need to pass the batch size when using distribution strategy... 

Fix to remove the need to pass the batch size when using distribution strategy when using predict on a numpy array if the number of samples is less than the default batch size.

PiperOrigin-RevId: 221523550
parent bd55d0de
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