[Mlir-commits] [mlir] [mlir][linalg] Add TransposeConv2D Pass (PR #68567)
Nicolas Vasilache
llvmlistbot at llvm.org
Fri Oct 20 09:28:15 PDT 2023
================
@@ -65,6 +65,10 @@ std::unique_ptr<Pass> createLinalgGeneralizationPass();
/// work on primitive types, if possible.
std::unique_ptr<Pass> createLinalgDetensorizePass();
+/// Create a pass to convert linalg.conv_2d_nhwc_fhwc(_q) to
+/// linalg.conv_2d_nhwc_hwcf(_q).
+std::unique_ptr<Pass> createLinalgTransposeConv2DPass();
----------------
nicolasvasilache wrote:
Can we avoid adding a very specific pass here and instead refactor to just have a new transform op ?
Downstream projects are free to wrap this into passes.
https://github.com/llvm/llvm-project/pull/68567
More information about the Mlir-commits
mailing list