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>
Loading
Please sign in to comment