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>