Skip to content
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

convolution #549

Open
vnatesh opened this issue Apr 13, 2023 · 2 comments
Open

convolution #549

vnatesh opened this issue Apr 13, 2023 · 2 comments

Comments

@vnatesh
Copy link

vnatesh commented Apr 13, 2023

Is it possible to generate code for convolution operations? For example, for convolution with stride s, the tensor algebra expression O(m,h,w) = F(m,k,i,j) * I(k, hs + i, ws + j) is currently invalid in the online tool

@rootjalex
Copy link

A paper that built on TACO (and I believe appears in MLSys this year) has support for this type of convolution, the github repo is here: https://github.com/nullplay/Unified-Convolution-Framework

@vnatesh
Copy link
Author

vnatesh commented Apr 21, 2023

Thank you for the reference Alex!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants