Commit 3ba1f72f authored by vihanjain's avatar vihanjain Committed by Gunhan Gulsoy
Browse files

Pull request for fixing warm-starting device placement (#17312) 

* Update checkpoint_utils.py

Fix device allocation bug for warm-starting op

* Update checkpoint_utils_test.py

Fix test
parent 7f53659b
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