[PATCH] D144108: [mlir][LinAlg][Transform] Add a transform op for conv2d to im2col
Mahesh Ravishankar via Phabricator via llvm-commits
llvm-commits at lists.llvm.org
Wed Feb 15 10:55:15 PST 2023
mravishankar requested changes to this revision.
mravishankar added a comment.
I know this pattern exists in IREE, but it is a bit of a hack. The representation of the im2col as a linalg.generic doesnt work as well. In reality it is similar to a gather. If this is upstreamed, it might be worth doing this right and not use an unnecessarily higher-dimensional operation for representing the im2col. Maybe we can chat offline?
Repository:
rG LLVM Github Monorepo
CHANGES SINCE LAST ACTION
https://reviews.llvm.org/D144108/new/
https://reviews.llvm.org/D144108
More information about the llvm-commits
mailing list