Commit b71b6b8c authored by Yong Tang's avatar Yong Tang Committed by Derek Murray
Browse files

Shape validation with random/shuffle related Dataset ops (#18682) 



* Add shape check for CacheDataset

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>

* Add shape check for ShuffleAndRepeatDataset

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>

* Add check for ShuffleDataset

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>

* Add shape check for RandomDataset

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>

* Add RangeDataset shape check

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>

* Sanitize with clang-format -i --style=Google

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>
parent 50f6683c
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment