Skip to content

Commit 1b11755

Browse files
rsudermanmemfrob
authored andcommitted
[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
1 parent 7ae1381 commit 1b11755

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

mlir/include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOpsSpec.tc

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -341,7 +341,7 @@ order of (`N`, `F`, `D`, `H`, `W`, `KD`, `KH`, `KW`, `C`).
341341
K(kd, kh, kw, c, f)));
342342
}
343343

344-
ods_def<PoolingNHWCSumOp>:
344+
ods_def<PoolingNHWCSumFOp>:
345345
def pooling_nhwc_sum
346346
(I: f32(N, H, W, C), K: f32(KH, KW))
347347
-> (O: f32(N, OH, OW, C))
@@ -352,7 +352,7 @@ def pooling_nhwc_sum
352352
ow * strides[1] + kw * dilations[1], c));
353353
}
354354

355-
ods_def<PoolingNHWCMaxOp>:
355+
ods_def<PoolingNHWCMaxFOp>:
356356
def pooling_nhwc_max
357357
(I: f32(N, H, W, C), K: f32(KH, KW))
358358
-> (O: f32(N, OH, OW, C))
@@ -367,7 +367,7 @@ def pooling_nhwc_max
367367
O(n, oh, ow, c));
368368
}
369369

370-
ods_def<PoolingNHWCMinOp>:
370+
ods_def<PoolingNHWCMinFOp>:
371371
def pooling_nhwc_min
372372
(I: f32(N, H, W, C), K: f32(KH, KW))
373373
-> (O: f32(N, OH, OW, C))

0 commit comments

Comments
 (0)