diff options
author | Allen Lavoie <allenl@google.com> | 2018-05-03 13:22:33 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-03 13:44:41 -0700 |
commit | 5a64e609d0eb94244067f5d7514605863c9f37c3 (patch) | |
tree | dbfe0ff8e2aa3ff788beabef757e90f4c2dbe4df /tensorflow/contrib/checkpoint/__init__.py | |
parent | 41dcb67efd272e9ce0e5071433f42a9d540ec6dc (diff) |
Checkpointable: Utilities to read object metadata
Useful for inspecting checkpoints programatically (e.g. in unit tests).
PiperOrigin-RevId: 195300780
Diffstat (limited to 'tensorflow/contrib/checkpoint/__init__.py')
-rw-r--r-- | tensorflow/contrib/checkpoint/__init__.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/contrib/checkpoint/__init__.py b/tensorflow/contrib/checkpoint/__init__.py index 1192cc44a1..d2c30f1215 100644 --- a/tensorflow/contrib/checkpoint/__init__.py +++ b/tensorflow/contrib/checkpoint/__init__.py @@ -16,7 +16,9 @@ For creating and managing dependencies: +@@CheckpointableObjectGraph @@dot_graph_from_checkpoint +@@object_metadata @@split_dependency """ @@ -26,6 +28,8 @@ from __future__ import print_function from tensorflow.contrib.checkpoint.python.split_dependency import split_dependency from tensorflow.contrib.checkpoint.python.visualize import dot_graph_from_checkpoint +from tensorflow.core.protobuf.checkpointable_object_graph_pb2 import CheckpointableObjectGraph +from tensorflow.python.training.checkpointable_utils import object_metadata from tensorflow.python.util.all_util import remove_undocumented |