diff options
author | Allen Lavoie <allenl@google.com> | 2018-06-14 09:40:33 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-14 09:43:03 -0700 |
commit | 5001a3f25bf709159b8cd40d3024885ff382acc3 (patch) | |
tree | a84c2bb0c016df9e2111630e03a0a084f4612f89 /tensorflow/contrib/checkpoint | |
parent | 3d5fa1f7f85e8cbd39227e921960fa36539ba3cd (diff) |
Add tf.contrib.checkpoint.list_objects for listing all Python dependencies of a
checkpointable object
Useful for asserting that all expected objects have been added as dependencies
in a unit test.
PiperOrigin-RevId: 200569520
Diffstat (limited to 'tensorflow/contrib/checkpoint')
-rw-r--r-- | tensorflow/contrib/checkpoint/__init__.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/contrib/checkpoint/__init__.py b/tensorflow/contrib/checkpoint/__init__.py index 8ae493ba99..257e93d283 100644 --- a/tensorflow/contrib/checkpoint/__init__.py +++ b/tensorflow/contrib/checkpoint/__init__.py @@ -16,6 +16,7 @@ Visualization and inspection: @@dot_graph_from_checkpoint +@@list_objects @@object_metadata Managing dependencies: @@ -42,9 +43,9 @@ 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 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__) - |