Commit 2addad84 authored by Yong Tang's avatar Yong Tang Committed by Rasmus Munk Larsen
Browse files

Validate the shape of count for RepeatDataset (#18150) 



* Validate the shape of count for RepeatDataset

The count in RepeatDataset requires a scalar though
the validation is only done in ops, not in shape function.

This fix validates the shape of count for RepeatDataset.

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

* Sanitize with clang-foramt -i

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

* Add test case for shape validation of count with RepeatDataset

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>
parent f16b50e3
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