Commit 4a1d1c84 authored by Yong Tang's avatar Yong Tang
Browse files

Fix mismatch of shape restriction in DrawBoundingBoxes



In the kernel of DrawBoundingBoxes, the shape of the input
images should be 4-D. Though in the shape function,
at the end `UnchangedShapeWithRankAtLeast(c, 3)` was used instead
(at the beginning of the shape function the validation is
`WithRank(c->input(0), 4, &images)` which is correct).

This fix address the discrepancy by changing to `UnchangedShape`.

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>
parent baa794b4
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment