taco icon indicating copy to clipboard operation
taco copied to clipboard

convolution

Open vnatesh opened this issue 1 year ago • 2 comments

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

vnatesh avatar Apr 13 '23 12:04 vnatesh

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

rootjalex avatar Apr 20 '23 22:04 rootjalex

Thank you for the reference Alex!

vnatesh avatar Apr 21 '23 14:04 vnatesh