[all-commits] [llvm/llvm-project] 068bf9: [mlir][linalg] Define a depthwise 2-D convolution op
Lei Zhang via All-commits
all-commits at lists.llvm.org
Tue Feb 9 05:56:27 PST 2021
Branch: refs/heads/main
Home: https://github.com/llvm/llvm-project
Commit: 068bf9e80220329f7baeaf8df34fb4818ad8aa58
https://github.com/llvm/llvm-project/commit/068bf9e80220329f7baeaf8df34fb4818ad8aa58
Author: Lei Zhang <antiagainst at google.com>
Date: 2021-02-09 (Tue, 09 Feb 2021)
Changed paths:
M mlir/include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOpsSpec.tc
M mlir/test/Dialect/Linalg/generalize-named-ops.mlir
A mlir/test/Dialect/Linalg/named-ops.mlir
Log Message:
-----------
[mlir][linalg] Define a depthwise 2-D convolution op
This commit defines linalg.depthwise_conv_2d_nhwc for depthwise
2-D convolution with NHWC input/output data format.
This op right now only support channel multiplier == 1, which is
the most common case.
Reviewed By: nicolasvasilache
Differential Revision: https://reviews.llvm.org/D94966
More information about the All-commits
mailing list