[Mlir-commits] [mlir] 12c3bca - [mlir] Rename linalg.pooling operations to have a FOp postfix for floating point
Rob Suderman
llvmlistbot at llvm.org
Fri Apr 2 12:00:17 PDT 2021
Author: Rob Suderman
Date: 2021-04-02T11:54:20-07:00
New Revision: 12c3bca1934a5eb0e23f6265d8fcc11ad6f4dea5
URL: https://github.com/llvm/llvm-project/commit/12c3bca1934a5eb0e23f6265d8fcc11ad6f4dea5
DIFF: https://github.com/llvm/llvm-project/commit/12c3bca1934a5eb0e23f6265d8fcc11ad6f4dea5.diff
LOG: [mlir] Rename linalg.pooling operations to have a FOp postfix for floating point
Linalg pooling operations only support floating point currently but integer
variants will soon be needed. Renaming to uncluse a FOp postfix to clarify.
Differential Revision: https://reviews.llvm.org/D99779
Added:
Modified:
mlir/include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOpsSpec.tc
Removed:
################################################################################
diff --git a/mlir/include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOpsSpec.tc b/mlir/include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOpsSpec.tc
index 66c7978d4252..ae29a1fe6fd9 100644
--- a/mlir/include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOpsSpec.tc
+++ b/mlir/include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOpsSpec.tc
@@ -341,7 +341,7 @@ order of (`N`, `F`, `D`, `H`, `W`, `KD`, `KH`, `KW`, `C`).
K(kd, kh, kw, c, f)));
}
-ods_def<PoolingNHWCSumOp>:
+ods_def<PoolingNHWCSumFOp>:
def pooling_nhwc_sum
(I: f32(N, H, W, C), K: f32(KH, KW))
-> (O: f32(N, OH, OW, C))
@@ -352,7 +352,7 @@ def pooling_nhwc_sum
ow * strides[1] + kw * dilations[1], c));
}
-ods_def<PoolingNHWCMaxOp>:
+ods_def<PoolingNHWCMaxFOp>:
def pooling_nhwc_max
(I: f32(N, H, W, C), K: f32(KH, KW))
-> (O: f32(N, OH, OW, C))
@@ -367,7 +367,7 @@ def pooling_nhwc_max
O(n, oh, ow, c));
}
-ods_def<PoolingNHWCMinOp>:
+ods_def<PoolingNHWCMinFOp>:
def pooling_nhwc_min
(I: f32(N, H, W, C), K: f32(KH, KW))
-> (O: f32(N, OH, OW, C))
More information about the Mlir-commits
mailing list