add assert_element_shape method for tf.contrib.data (#17480)
authorYan Facai (颜发才) <facai.yan@gmail.com>
Fri, 6 Apr 2018 02:51:12 +0000 (10:51 +0800)
committerDerek Murray <derek.murray@gmail.com>
Fri, 6 Apr 2018 02:51:12 +0000 (19:51 -0700)
commitd82b2f71b60d5fff48884c20c7b85e517330e91f
tree6bd9f72261886bee00f016f286b7c9488312a68c
parenta1fbdf8268d8043190fe7fa0b3c5796a455ea529
add assert_element_shape method for tf.contrib.data (#17480)

* ENH: add assert_element_shape method

* CLN: add indentation

* ENH: raise exception when wrong shape is given

* CLN: fix too long line
tensorflow/contrib/data/__init__.py
tensorflow/contrib/data/python/kernel_tests/BUILD
tensorflow/contrib/data/python/kernel_tests/batch_dataset_op_test.py
tensorflow/contrib/data/python/ops/BUILD
tensorflow/contrib/data/python/ops/batching.py