[Mlir-commits] [mlir] [mlir][linalg] Add TransposeConv2D Pass (PR #68567)

Jack Frankland llvmlistbot at llvm.org
Mon Oct 23 08:39:38 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();
----------------
FranklandJack wrote:

Ah that's a good idea. I didn't think of this before but makes sense - I've updated the patch to introduce a transform op and updated the tests to check the behaviour is correct going through the transform dialect interpreter.

https://github.com/llvm/llvm-project/pull/68567


More information about the Mlir-commits mailing list