New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Refactor type conversion for read back #7044
Conversation
d9e4e0c
to
1a59012
Compare
dcc1463
to
3a4e1f4
Compare
@qjia7 @xhcao @haoyunfeix @gyagp PTAL |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Based on #6965's description, it seems that ResizeNearestNeighbor doesn't create the right TypedArray buffer based on dType
. Should we fix it there or do the post processing to cast the data to the right type when reading data back like other backends? @pyu10055 @mattsoulanille
@pyu10055 @mattsoulanille, PTAL This PR can fix DeepLabV3(in e2e) failure. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks @mattsoulanille. |
27efe53
to
a74bd16
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Reviewed 5 of 5 files at r1.
Reviewable status: complete! 2 of 1 approvals obtained
* Refactor type conversion for read back Bug: tensorflow#6965 * Cleanup Co-authored-by: Ping Yu <4018+pyu10055@users.noreply.github.com>
Bug: #6965
This change is