diff options
| author |  Allen Lavoie <allenl@google.com> | 2018-06-15 17:07:57 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-15 17:10:43 -0700 |
| commit | f6148d7a4e2d080da93d21de2f13b601465c7528 (patch) | |
| tree | 12329f7448ebd9735de355211e831674e24d6ca5 /tensorflow/contrib/checkpoint | |
| parent | d1daba6ac82461cd64dc070534bc613a70527520 (diff) | |
Add tf.contrib.checkpoint.CheckpointableBase for isinstance checks.
(Also planning to use this in Sonnet)
PiperOrigin-RevId: 200797385
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 9aa4614967..38856417c0 100644 --- a/tensorflow/contrib/checkpoint/__init__.py +++ b/tensorflow/contrib/checkpoint/__init__.py @@ -22,6 +22,7 @@ Visualization and inspection: Managing dependencies: @@capture_dependencies @@Checkpointable +@@CheckpointableBase @@CheckpointableObjectGraph @@NoDependency @@split_dependency @@ -41,6 +42,7 @@ from tensorflow.contrib.checkpoint.python.split_dependency import split_dependen 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.base import Checkpointable +from tensorflow.python.training.checkpointable.base import CheckpointableBase 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 @@ -51,4 +53,3 @@ from tensorflow.python.training.checkpointable.util import object_metadata from tensorflow.python.util.all_util import remove_undocumented remove_undocumented(module_name=__name__) - |