-
Notifications
You must be signed in to change notification settings - Fork 3k
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
Optimize the resize and upsample #1426
Conversation
thanks! do you mind to share numbers if any for this improvement please? |
Added in the PR description section. |
Can we add one more very simple change, that being to check if any work is required? e.g. ssd_mobilenet in mlperf has a resize that does nothing as the input size matches the resize, but nothing checks if all the scales values are 1.0. |
Just saw the comment. Adding it in a new PR. |
Description: Describe your changes.
Optimize the resize and upsample operators
Motivation and Context
For case with input with shape [1,128, 267, 200] and scales [1, 1, 1.97, 2], Resize and upsample get 15x gain (w/o: 1020ms, w: 71ms on my local box). It should benefit other scenarios at similar level.