From 5a64e609d0eb94244067f5d7514605863c9f37c3 Mon Sep 17 00:00:00 2001 From: Allen Lavoie Date: Thu, 3 May 2018 13:22:33 -0700 Subject: Checkpointable: Utilities to read object metadata Useful for inspecting checkpoints programatically (e.g. in unit tests). PiperOrigin-RevId: 195300780 --- tensorflow/contrib/checkpoint/__init__.py | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'tensorflow/contrib/checkpoint/__init__.py') 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 -- cgit v1.2.3