projects / platform / upstream / llvm.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: ae69f46) | patch
[mlir][linalg] Add pooling_nchw_max, conv_2d_nchw as yaml ops.
authorYi Zhang <cathyzhyi@google.com>
Fri, 23 Jul 2021 16:15:21 +0000 (16:15 +0000)
committerTobias Gysi <gysit@google.com>
Fri, 23 Jul 2021 17:37:15 +0000 (17:37 +0000)
commitdeebf18512266e0e6917508052f6d9bbd06c7d5e
tree63c125cbfd546824d8a2a438fc7c07ec76b85bcdtree | snapshot
parentae69f46867cd99d18ec7e23509eb1d71a098d48fcommit | diff
[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
mlir/include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOps.yaml diff | blob | history
mlir/include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOpsSpec.tc diff | blob | history
mlir/lib/Dialect/Linalg/Transforms/Vectorization.cpp diff | blob | history
mlir/python/mlir/dialects/linalg/opdsl/ops/core_named_ops.py diff | blob | history
mlir/test/Dialect/Linalg/named-ops.mlir diff | blob | history
mlir/test/Integration/Dialect/Linalg/CPU/test-conv-2d-nchw-call.mlir diff | blob | history
Domain: System / Toolchain;