Commit a1fbdf82 authored by Yong Tang's avatar Yong Tang Committed by Andrew Harp
Browse files

Validate the shape of TakeDataset and SkipDataset (#18267) 



* Validate the shape of TakeDataset and SkipDataset

The `count` inputs of the TakeDataset and SkipDataset require
scalar. That was not validated before though.

This fix validates the shape of count for TakeDataset and SkipDataset.

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

* Add test case for invalide `count` of TakeDataset

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

* Validate shape of count for SkipDataset

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

* Add test case for invalid count shape for SkipDataset

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