-
Notifications
You must be signed in to change notification settings - Fork 625
[TOSA] Conv3d legalization #4383
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
Conversation
4042fcd to
d7fa53c
Compare
Lallapallooza
left a comment
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 PR, overall LGTM, few nits.
sahas3
left a comment
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 this substantial contribution.
sahas3
left a comment
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 addressing the feedback so quickly. It'll be great if you address the bias computation feedback. The rest are mainly nits.
2c0d4a2 to
25200e9
Compare
- Extend Torch to TOSA conversion with full aten.conv3d coverage, mapping padding/stride/dilation configs and bias handling into canonical TOSA conv ops. - Refactor TorchToTosa lowering utilities to share more code between conv variants. - Update conversion tests plus PT1 xfail lists to reflect the newly supported 3D convolution paths. Signed-off-by: Cathal Corbett <[email protected]> Change-Id: Iaf1261e121dec1ddb84b814e9058bb7ebadd4de7
Change-Id: Iaf1261e121dec1ddb84b814e9058bb7ebadd4de7