Shape validation with random/shuffle related Dataset ops (#18682)
authorYong Tang <yong.tang.github@outlook.com>
Thu, 19 Apr 2018 16:13:35 +0000 (09:13 -0700)
committerDerek Murray <derek.murray@gmail.com>
Thu, 19 Apr 2018 16:13:35 +0000 (09:13 -0700)
commitb71b6b8ca9ade8b39d77f0373210fe58dfccf4f4
treeef36e906b74a54d3e5de1435c23d4003c6ae9047
parent50f6683ca50e6d4e7008d6d1b437b407d6a62e92
Shape validation with random/shuffle related Dataset ops (#18682)

* Add shape check for CacheDataset

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

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

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

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

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