Pull request for fixing warm-starting device placement (#17312) (#17314)
* Update checkpoint_utils.py Fix device allocation bug for warm-starting op * Update checkpoint_utils_test.py Fix test
Loading
Please sign in to comment
* Update checkpoint_utils.py Fix device allocation bug for warm-starting op * Update checkpoint_utils_test.py Fix test