aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/training/__init__.py
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2017-02-13 19:16:30 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-02-13 19:28:31 -0800
commit582fc4a3bc02b160844c03c2155f206aa07d7e6e (patch)
tree2ebce20e8eecbd30ab849b198d2f4a5dac77a1ea /tensorflow/contrib/training/__init__.py
parentb71f11361e025c1245363a283a9761cd3368d3c9 (diff)
Fix broken links for unsealed contrib API guides
contrib.learn.monitors will not have an API guide, since it is deprecated and unsealed. Fixes sealing for tf.contrib.training. Change: 147424947
Diffstat (limited to 'tensorflow/contrib/training/__init__.py')
-rw-r--r--tensorflow/contrib/training/__init__.py20
1 files changed, 18 insertions, 2 deletions
diff --git a/tensorflow/contrib/training/__init__.py b/tensorflow/contrib/training/__init__.py
index 245f36c2fe..929f2be9be 100644
--- a/tensorflow/contrib/training/__init__.py
+++ b/tensorflow/contrib/training/__init__.py
@@ -23,6 +23,13 @@
@@weighted_resample
@@bucket
@@bucket_by_sequence_length
+@@GreedyLoadBalancingStrategy
+@@byte_size_load_fn
+@@FailureTolerator
+@@rejection_sample
+@@stratified_sample
+@@resample_at_rate
+@@weighted_resample
"""
from __future__ import absolute_import
@@ -49,7 +56,16 @@ from tensorflow.contrib.training.python.training.training import clip_gradient_n
from tensorflow.contrib.training.python.training.training import create_train_op
from tensorflow.contrib.training.python.training.training import multiply_gradients
from tensorflow.contrib.training.python.training.training import train
-from tensorflow.python.util.all_util import make_all
# pylint: enable=unused-import,wildcard-import
-__all__ = make_all(__name__)
+from tensorflow.python.util.all_util import remove_undocumented
+
+# Allow explicitly imported symbols. Symbols imported with * must also be
+# whitelisted here or in the module docstring above.
+_allowed_symbols = [
+ 'checkpoints_iterator', 'evaluate_once', 'evaluate_repeatedly',
+ 'get_or_create_eval_step', 'StopAfterNEvalsHook', 'SummaryAtEndHook',
+ 'wait_for_new_checkpoint', 'add_gradients_summaries', 'clip_gradient_norms',
+ 'create_train_op', 'multiply_gradients', 'train']
+
+remove_undocumented(__name__, _allowed_symbols)