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

[mlir][linalg] Add suffix "Op" to pooling TC ops. 

Reviewed By: antiagainst

Differential Revision: https://reviews.llvm.org/D97946
This commit is contained in:
Hanhan Wang 2021-03-04 07:08:19 -08:00
parent 2903934faa
commit b47c6c686c
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<PoolingNHWCSum>:
ods_def<PoolingNHWCSumOp>:
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<PoolingNHWCMax>:
ods_def<PoolingNHWCMaxOp>:
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<PoolingNHWCMin>:
ods_def<PoolingNHWCMinOp>:
def pooling_nhwc_min
(I: f32(N, H, W, C), K: f32(KH, KW))
-> (O: f32(N, OH, OW, C))