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

Add shape check for batch related Dataset ops (#18683) 



* Add shape check for PrefetchDataset

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

* Add BatchDataset shape check

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

* Add shape check for SlideDataset

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

* Add shape check for DenseToSparseBatchDataset

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 edb29d02
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