From 12c3bca1934a5eb0e23f6265d8fcc11ad6f4dea5 Mon Sep 17 00:00:00 2001 From: Rob Suderman Date: Thu, 1 Apr 2021 16:58:01 -0700 Subject: [PATCH] [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 --- .../mlir/Dialect/Linalg/IR/LinalgNamedStructuredOpsSpec.tc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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: +ods_def: 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: +ods_def: 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: +ods_def: def pooling_nhwc_min (I: f32(N, H, W, C), K: f32(KH, KW)) -> (O: f32(N, OH, OW, C))