aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/data/__init__.py
diff options
context:
space:
mode:
authorGravatar Scott Zhu <scottzhu@google.com>2018-04-13 17:52:20 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-04-13 17:57:27 -0700
commit3652556dab3ebfe0152232facc7304fe5754aecb (patch)
tree9a9cecde4c85dc53548a185f9bd6d7c6e0591262 /tensorflow/contrib/data/__init__.py
parentef24ad14502e992716c49fdd5c63e6b2c2fb6b5a (diff)
Merge changes from github.
PiperOrigin-RevId: 192850372
Diffstat (limited to 'tensorflow/contrib/data/__init__.py')
-rw-r--r--tensorflow/contrib/data/__init__.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/contrib/data/__init__.py b/tensorflow/contrib/data/__init__.py
index f58e5ec1f0..637b1dc46c 100644
--- a/tensorflow/contrib/data/__init__.py
+++ b/tensorflow/contrib/data/__init__.py
@@ -25,6 +25,7 @@ See the @{$datasets$Importing Data} Programmer's Guide for an overview.
@@Counter
@@SqlDataset
+@@assert_element_shape
@@batch_and_drop_remainder
@@bucket_by_sequence_length
@@dense_to_sparse_batch
@@ -55,6 +56,7 @@ from __future__ import print_function
# pylint: disable=unused-import
+from tensorflow.contrib.data.python.ops.batching import assert_element_shape
from tensorflow.contrib.data.python.ops.batching import batch_and_drop_remainder
from tensorflow.contrib.data.python.ops.batching import dense_to_sparse_batch
from tensorflow.contrib.data.python.ops.batching import map_and_batch