[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