maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

[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
This commit is contained in:
Rob Suderman 2021-04-01 16:58:01 -07:00
parent 944adbf285
commit 12c3bca193
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mlir/include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOpsSpec.tc
View File

@ -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))