Commit 4186d29d authored by Ruoxin Sang's avatar Ruoxin Sang Committed by TensorFlower Gardener
Browse files

Move training loop from device to host in TPUStrategy. 

PiperOrigin-RevId: 223461384
parent ecf27945
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