aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/training/BUILD
diff options
context:
space:
mode:
authorGravatar Eugene Brevdo <ebrevdo@google.com>2018-04-18 20:15:27 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-04-18 20:18:06 -0700
commitfd10bfb61ef6b1885c8fa2459522fa98305df703 (patch)
treedae10dcaf4a0411e270013528c6e314b24889645 /tensorflow/contrib/training/BUILD
parent81cabadc78811a216381fbf30715b1313684e32f (diff)
Expose tf.contrib.training.{prepend_from_queue_and_padded_batch_dataset}
Also its helper method "enqueue_in_queue_dataset". PiperOrigin-RevId: 193458095
Diffstat (limited to 'tensorflow/contrib/training/BUILD')
-rw-r--r--tensorflow/contrib/training/BUILD1
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/training/BUILD b/tensorflow/contrib/training/BUILD
index 4d2bfd3e43..5de55b5f7f 100644
--- a/tensorflow/contrib/training/BUILD
+++ b/tensorflow/contrib/training/BUILD
@@ -60,6 +60,7 @@ py_library(
"//tensorflow/python:util",
"//tensorflow/python:variable_scope",
"//tensorflow/python:variables",
+ "//tensorflow/python/data",
"//tensorflow/python/estimator:inputs_queues",
"//third_party/py/numpy",
"@six_archive//:six",