Commit 568127ac authored by Yong Tang's avatar Yong Tang Committed by Vijay Vasudevan
Browse files

Improve shape inference with `DecodeAndCropJpeg` (#13750)



* Improve shape inference with `DecodeAndCropJpeg`

While working on improving shape inference for several
other ops in 13561 and 13193, I noticed that `DecodeAndCropJpeg`
does not inference shape even though crop size might have already
be provided. In that case the shape will be `[h, w, channel]`
and `h`, `w` is part of the `crop_window`.

This fix updates the shape function in `DecodeAndCropJpeg`
for improving shape inference.

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

* Add test cases to cover shape inference for `DecodeAndCropJpeg`

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

* Address failed unit tests

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