[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