aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/BUILD
diff options
context:
space:
mode:
authorGravatar Allen Lavoie <allenl@google.com>2018-04-20 12:40:57 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-04-20 12:43:51 -0700
commit517d1912f4ec71180944320350a3694332a1dedc (patch)
tree23fd05eb7e72c8ce13766a516280622089cd0c70 /tensorflow/contrib/BUILD
parentb23e91d247368f2046dae035b5c7bdda56512077 (diff)
Add a utility to visualize object-based checkpoints
Useful for generating a warm fuzzy feeling that everything you think should be saved was saved, and for explaining what object-based checkpointing is. (Also useful on the former front will be a planned "assert that all of this Graph's trainable variables are accessible from object X" function.) Somewhat hacky since it generates strings rather than using the pydot bindings (and so works without a pydot dependency). PiperOrigin-RevId: 193708003
Diffstat (limited to 'tensorflow/contrib/BUILD')
-rw-r--r--tensorflow/contrib/BUILD1
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/BUILD b/tensorflow/contrib/BUILD
index 7e47516550..d28392a62c 100644
--- a/tensorflow/contrib/BUILD
+++ b/tensorflow/contrib/BUILD
@@ -25,6 +25,7 @@ py_library(
"//tensorflow/contrib/batching:batch_py",
"//tensorflow/contrib/bayesflow:bayesflow_py",
"//tensorflow/contrib/boosted_trees:init_py",
+ "//tensorflow/contrib/checkpoint/python:checkpoint",
"//tensorflow/contrib/cloud:cloud_py",
"//tensorflow/contrib/cluster_resolver:cluster_resolver_pip",
"//tensorflow/contrib/cluster_resolver:cluster_resolver_py",