-
Notifications
You must be signed in to change notification settings - Fork 687
Arm backend: Add TOSA (DW) Conv2d dialect operator #14843
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
base: main
Are you sure you want to change the base?
Arm backend: Add TOSA (DW) Conv2d dialect operator #14843
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/14843
Note: Links to docs will display an error until the docs builds have been completed. ❌ 4 New Failures, 4 Unrelated FailuresAs of commit 89af054 with merge base 4c064e3 ( NEW FAILURES - The following jobs have failed:
BROKEN TRUNK - The following jobs failed but were present on the merge base:👉 Rebase onto the `viable/strict` branch to avoid these failures
This comment was automatically generated by Dr. CI and updates every 15 minutes. |
Adds backend dialect operators for: - Conv2D - Depthwise Conv2D Also removes AddBiasPass and moves that logic to new RewriteConv2dPass. Signed-off-by: Oscar Andersson <oscar.andersson@arm.com> Change-Id: I1185f51ded5e931ca042f445934cd21e20fdf469
26268d0
to
98d31d8
Compare
from executorch.backends.arm.tosa.specification import Tosa_1_00, TosaSpecification | ||
from executorch.backends.arm.tosa.utils import tosa_shape | ||
|
||
|
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 for the rename :)
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! I like this tosa op abstraction layer direction :)
Adds backend dialect operators for:
Also removes AddBiasPass and moves that logic to new RewriteConv2dPass.
cc @freddan80 @per @zingo @digantdesai