Checkpointable: Re-use the Saver's SaveableObject infrastructure to create
authorAllen Lavoie <allenl@google.com>
Thu, 22 Feb 2018 20:01:57 +0000 (12:01 -0800)
committerTensorFlower Gardener <gardener@tensorflow.org>
Thu, 22 Feb 2018 20:05:52 +0000 (12:05 -0800)
commita4de23973ddddfa8dc26d846dc0a902942347b11
tree6fbbd3f57e1736b1ad0a649f631d00fd213407c6
parent725d049caea2b75eae373760127ed9b55138f7dc
Checkpointable: Re-use the Saver's SaveableObject infrastructure to create
restore ops, cache them.

The basic infrastructure is usable after this CL in graph and eager, but I still need to make a bunch of objects Checkpointable and make some other usability fixes.

Also sets some of the groundwork for feeding and fetching Python values during save/restore (as in, save has a feed dict now; gathering feeds and placeholders from Checkpointable objects is still to do but should be relatively straightforward).

PiperOrigin-RevId: 186652696
tensorflow/contrib/eager/python/checkpointable_utils.py
tensorflow/contrib/eager/python/checkpointable_utils_test.py
tensorflow/python/BUILD
tensorflow/python/ops/variables.py
tensorflow/python/training/checkpointable.py
tensorflow/python/training/checkpointable_utils.py [new file with mode: 0644]