[all-commits] [llvm/llvm-project] 4a3d20: [mlir][linalg] Add TransposeConv2D Transform Op (#...
Jack Frankland via All-commits
all-commits at lists.llvm.org
Tue Nov 28 01:56:27 PST 2023
Branch: refs/heads/main
Home: https://github.com/llvm/llvm-project
Commit: 4a3d2088d61e994dd1aa4e66cdecd15f28d2f397
https://github.com/llvm/llvm-project/commit/4a3d2088d61e994dd1aa4e66cdecd15f28d2f397
Author: Jack Frankland <jack.frankland at arm.com>
Date: 2023-11-28 (Tue, 28 Nov 2023)
Changed paths:
M mlir/include/mlir/Dialect/Linalg/TransformOps/LinalgTransformOps.td
M mlir/include/mlir/Dialect/Linalg/Transforms/Transforms.h
M mlir/lib/Dialect/Linalg/TransformOps/LinalgTransformOps.cpp
M mlir/lib/Dialect/Linalg/Transforms/CMakeLists.txt
A mlir/lib/Dialect/Linalg/Transforms/TransposeConv2D.cpp
A mlir/test/Dialect/Linalg/transpose-conv2d.mlir
Log Message:
-----------
[mlir][linalg] Add TransposeConv2D Transform Op (#68567)
* Add a LinAlg pass to convert 2D convolutions and quantized 2D
convolutions that have the `FHWC` filter channel ordering into a
transpose followed by 2D convolutions that have the `HWCF` channel
ordering.
* Add a lit test to check the semantics of the transformation are
correct for both quantized and unquantized variants.
Signed-off-by: Jack Frankland <jack.frankland at arm.com>
More information about the All-commits
mailing list