diff options
author | Allen Lavoie <allenl@google.com> | 2018-04-19 09:54:40 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-19 09:57:23 -0700 |
commit | 436f1434060d7f370baae9661baacc6cf27415ec (patch) | |
tree | 9276bf419abb2e1bcfccdb7c43bea7ffbab80283 /tensorflow/contrib/__init__.py | |
parent | c173157bdc132460c6f424a9803221e74fc73f59 (diff) |
Create a skeleton tf.contrib.checkpoint.
My plan for this is to incubate tools for working with object-based checkpoints:
- Tools for managing dependency graphs, e.g. checkpointable lists/dictionaries
- Inspecting/visualizing checkpoints
- Listing variables and gathering initializers from a Checkpointable object
and its dependencies
- Verifying all variables are accessible as dependencies, which should make
converting existing graph building Saver uses easier/safer.
This CL includes none of those things, it just moves the split_dependency tool
here instead of contrib/eager.
PiperOrigin-RevId: 193531292
Diffstat (limited to 'tensorflow/contrib/__init__.py')
-rw-r--r-- | tensorflow/contrib/__init__.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/__init__.py b/tensorflow/contrib/__init__.py index 36cc5144d0..0d163daa6e 100644 --- a/tensorflow/contrib/__init__.py +++ b/tensorflow/contrib/__init__.py @@ -24,6 +24,7 @@ import os # Add projects here, they will show up under tf.contrib. from tensorflow.contrib import batching from tensorflow.contrib import bayesflow +from tensorflow.contrib import checkpoint from tensorflow.contrib import cloud from tensorflow.contrib import cluster_resolver from tensorflow.contrib import coder |