projects / platform / upstream / tensorflow.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: faa09ad) | patch
Checkpointable: Fix device placement when restoring name-based checkpoints.
authorAllen Lavoie <allenl@google.com>
Wed, 7 Mar 2018 23:52:25 +0000 (15:52 -0800)
committerTensorFlower Gardener <gardener@tensorflow.org>
Wed, 7 Mar 2018 23:56:33 +0000 (15:56 -0800)
commit5ae2d41e7a1daf4b00b24dda683fabf7c283df7c
treed3f1e30179d5fec2944c85ac4e4a5bfe9a401c7dtree | snapshot
parentfaa09ad9d3eb9f7a4dcd7c11f3b1e22e13496afdcommit | diff
Checkpointable: Fix device placement when restoring name-based checkpoints.

Just need to put the restore ops on a CPU.

PiperOrigin-RevId: 188248198
tensorflow/contrib/eager/python/checkpointable_utils.py diff | blob | history
tensorflow/contrib/eager/python/checkpointable_utils_test.py diff | blob | history
Domain: Machine Learning / ML Framework;