Commit 5ae2d41e authored by Allen Lavoie's avatar Allen Lavoie Committed by TensorFlower Gardener
Browse files

Checkpointable: Fix device placement when restoring name-based checkpoints. 

Just need to put the restore ops on a CPU.

PiperOrigin-RevId: 188248198
parent faa09ad9
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