[all-commits] [llvm/llvm-project] c8e056: [mlir][linalg] Add channel-first variants of convo...

[Kon via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: c8e056065898bba2d51fb655da40bd4ee7beedcf
      https://github.com/llvm/llvm-project/commit/c8e056065898bba2d51fb655da40bd4ee7beedcf
  Author: kon72 <kinsei0916 at gmail.com>
  Date:   2023-05-12 (Fri, 12 May 2023)

  Changed paths:
    M mlir/include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOps.yaml
    M mlir/python/mlir/dialects/linalg/opdsl/ops/core_named_ops.py
    M mlir/test/Dialect/Linalg/named-ops.mlir

  Log Message:
  -----------
  [mlir][linalg] Add channel-first variants of convolution

This change adds the following three operations and unit tests for them:

- conv_3d_ncdhw_fcdhw
- depthwise_conv_1d_ncw_cw
- depthwise_conv_3d_ncdhw_cdhw

Reviewed By: nicolasvasilache

Differential Revision: https://reviews.llvm.org/D150054