Add shape check for batch related Dataset ops (#18683)
authorYong Tang <yong.tang.github@outlook.com>
Thu, 19 Apr 2018 16:13:21 +0000 (09:13 -0700)
committerDerek Murray <derek.murray@gmail.com>
Thu, 19 Apr 2018 16:13:21 +0000 (09:13 -0700)
commit50f6683ca50e6d4e7008d6d1b437b407d6a62e92
tree9264ef8bd60e0c422f45da31a17c2161cf4b19dc
parentedb29d02765b45e712042725dc06b65f5e610327
Add shape check for batch related Dataset ops (#18683)

* Add shape check for PrefetchDataset

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Add BatchDataset shape check

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Add shape check for SlideDataset

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Add shape check for DenseToSparseBatchDataset

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Sanitize with clang-format -i --style=Google

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
tensorflow/core/ops/dataset_ops.cc