[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