Add pass to convert nchw convolutions to nhwc (iree-org#15)

Include partial support for propagating and cancelling transposes generated when converting from nchw to nhwc
nirvedhmeshram · Sep 30, 2022 · 0b00f3a · 0b00f3a
1 parent 7d1d9c1
commit 0b00f3a
Show file tree

Hide file tree

Showing 6 changed files with 711 additions and 0 deletions.
diff --git a/compiler/src/iree/compiler/Dialect/Flow/Transforms/BUILD b/compiler/src/iree/compiler/Dialect/Flow/Transforms/BUILD
@@ -36,6 +36,7 @@ iree_compiler_cc_library(
         "CleanupTensorShapes.cpp",
         "ConvertConv2D1x1ToMatmul.cpp",
         "ConvertConv2DToImg2Col.cpp",
+        "ConvertConvNchwToNhwc.cpp",
         "ConvertLinalgMatmulToMmt4D.cpp",
         "ConvertRegionToWorkgroups.cpp",
         "ConvertToFlow.cpp",

diff --git a/compiler/src/iree/compiler/Dialect/Flow/Transforms/CMakeLists.txt b/compiler/src/iree/compiler/Dialect/Flow/Transforms/CMakeLists.txt
@@ -32,6 +32,7 @@ iree_cc_library(
     "CaptureDispatchDynamicDims.cpp"
     "CleanupNumericNarrowing.cpp"
     "CleanupTensorShapes.cpp"
+    "ConvertConvNchwToNhwc.cpp"
     "ConvertConv2D1x1ToMatmul.cpp"
     "ConvertConv2DToImg2Col.cpp"
     "ConvertLinalgMatmulToMmt4D.cpp"