diff options
author | 2018-05-24 10:30:41 -0700 | |
---|---|---|
committer | 2018-05-24 10:35:32 -0700 | |
commit | 53cd5c01407451cf918c1d1c1f5ca640b7d5dbc8 (patch) | |
tree | 593bb7990d21a3cf7057d9eead2bf571f6014ca7 /tensorflow/contrib/checkpoint | |
parent | 1403625e860ffb8fb9af1bc75c1cea8f73e0478e (diff) |
Add a checkpointable map data structure
PiperOrigin-RevId: 197913890
Diffstat (limited to 'tensorflow/contrib/checkpoint')
-rw-r--r-- | tensorflow/contrib/checkpoint/__init__.py | 3 | ||||
-rw-r--r-- | tensorflow/contrib/checkpoint/python/BUILD | 5 |
2 files changed, 7 insertions, 1 deletions
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( |