taco
taco copied to clipboard
convolution
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
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
Thank you for the reference Alex!