diff options
author | Allen Lavoie <allenl@google.com> | 2018-06-14 12:02:32 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-14 12:09:39 -0700 |
commit | 7ccf1937b863a7f5cfb5d159d44671138d7393bf (patch) | |
tree | f33c446cfc05d13151bf22efe7a5a07eaf94f5e2 /tensorflow/contrib/checkpoint | |
parent | 91ec6cc4943f5500453cb09dc7ccdc265722312b (diff) |
Factor a "capture_dependencies" scope out of Template.
I don't intend for this to get used much directly, but it's handy for Template-like frameworks (e.g. Sonnet), to let them re-enter the dependency-capturing part of Templates.
PiperOrigin-RevId: 200595624
Diffstat (limited to 'tensorflow/contrib/checkpoint')
-rw-r--r-- | tensorflow/contrib/checkpoint/__init__.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/contrib/checkpoint/__init__.py b/tensorflow/contrib/checkpoint/__init__.py index 257e93d283..9aa4614967 100644 --- a/tensorflow/contrib/checkpoint/__init__.py +++ b/tensorflow/contrib/checkpoint/__init__.py @@ -20,6 +20,7 @@ Visualization and inspection: @@object_metadata Managing dependencies: +@@capture_dependencies @@Checkpointable @@CheckpointableObjectGraph @@NoDependency @@ -43,9 +44,11 @@ from tensorflow.python.training.checkpointable.base import Checkpointable from tensorflow.python.training.checkpointable.base import NoDependency from tensorflow.python.training.checkpointable.data_structures import List from tensorflow.python.training.checkpointable.data_structures import Mapping +from tensorflow.python.training.checkpointable.util import capture_dependencies from tensorflow.python.training.checkpointable.util import list_objects from tensorflow.python.training.checkpointable.util import object_metadata from tensorflow.python.util.all_util import remove_undocumented remove_undocumented(module_name=__name__) + |