[all-commits] [llvm/llvm-project] deebf1: [mlir][linalg] Add pooling_nchw_max, conv_2d_nchw ...

[Yi Zhang via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: deebf18512266e0e6917508052f6d9bbd06c7d5e
      https://github.com/llvm/llvm-project/commit/deebf18512266e0e6917508052f6d9bbd06c7d5e
  Author: Yi Zhang <cathyzhyi at google.com>
  Date:   2021-07-23 (Fri, 23 Jul 2021)

  Changed paths:
    M mlir/include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOps.yaml
    M mlir/include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOpsSpec.tc
    M mlir/lib/Dialect/Linalg/Transforms/Vectorization.cpp
    M mlir/python/mlir/dialects/linalg/opdsl/ops/core_named_ops.py
    M mlir/test/Dialect/Linalg/named-ops.mlir
    M mlir/test/Integration/Dialect/Linalg/CPU/test-conv-2d-nchw-call.mlir

  Log Message:
  -----------
  [mlir][linalg] Add pooling_nchw_max, conv_2d_nchw as yaml ops.

- Add pooling_nchw_max.
- Move conv_2d_nchw to yaml ops and add strides and dilation attributes.

Reviewed By: gysit

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