From 53cd5c01407451cf918c1d1c1f5ca640b7d5dbc8 Mon Sep 17 00:00:00 2001 From: Allen Lavoie Date: Thu, 24 May 2018 10:30:41 -0700 Subject: Add a checkpointable map data structure PiperOrigin-RevId: 197913890 --- tensorflow/contrib/checkpoint/__init__.py | 3 +++ tensorflow/contrib/checkpoint/python/BUILD | 5 ++++- 2 files changed, 7 insertions(+), 1 deletion(-) (limited to 'tensorflow/contrib/checkpoint') diff --git a/tensorflow/contrib/checkpoint/__init__.py b/tensorflow/contrib/checkpoint/__init__.py index bd0bc9e56b..8ae493ba99 100644 --- a/tensorflow/contrib/checkpoint/__init__.py +++ b/tensorflow/contrib/checkpoint/__init__.py @@ -26,6 +26,7 @@ Managing dependencies: Checkpointable data structures: @@List +@@Mapping @@UniqueNameTracker """ @@ -40,8 +41,10 @@ from tensorflow.core.protobuf.checkpointable_object_graph_pb2 import Checkpointa 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 object_metadata from tensorflow.python.util.all_util import remove_undocumented remove_undocumented(module_name=__name__) + diff --git a/tensorflow/contrib/checkpoint/python/BUILD b/tensorflow/contrib/checkpoint/python/BUILD index 0b67619c11..7b200a29bf 100644 --- a/tensorflow/contrib/checkpoint/python/BUILD +++ b/tensorflow/contrib/checkpoint/python/BUILD @@ -20,7 +20,10 @@ py_library( srcs = ["containers.py"], srcs_version = "PY2AND3", visibility = ["//tensorflow:internal"], - deps = ["//tensorflow/python/training/checkpointable:base"], + deps = [ + "//tensorflow/python/training/checkpointable:base", + "//tensorflow/python/training/checkpointable:data_structures", + ], ) py_test( -- cgit v1.2.3