diff --git a/mlir/include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOpsSpec.tc b/mlir/include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOpsSpec.tc --- a/mlir/include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOpsSpec.tc +++ b/mlir/include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOpsSpec.tc @@ -341,7 +341,7 @@ 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 @@ 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 @@ 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))