[Mlir-commits] [mlir] [mlir][linalg] Implement Winograd Conv2D. (PR #94470)
Oleksandr Alex Zinenko
llvmlistbot at llvm.org
Fri Jun 14 11:37:30 PDT 2024
================
@@ -154,4 +154,96 @@ def Linalg_SoftmaxOp : Linalg_Op<"softmax",
let hasVerifier = 1;
}
+def Linalg_WinogradFilterTransformOp : Linalg_Op<"winograd_filter_transform",
+ [DeclareOpInterfaceMethods<TilingInterface,
+ ["getIterationDomain",
+ "getLoopIteratorTypes",
+ "getResultTilePosition",
+ "getTiledImplementation"]>]> {
+ let summary = "Winograd filter transform operator";
+ let description = [{
+ linalg.winograd_filter_transform transforms the filter of conv2D.
----------------
ftynse wrote:
Please provide a more extended description, pseudo-code showing computation before/after or equivalent loop-level IR is welcome here.
https://github.com/llvm/llvm-project/pull/94470
More information about the Mlir-commits
mailing list