Skip to content

Commit 92abf7c

Browse files
Abhishek-VarmaAbhishek-Varma
authored
[NFC][Linalg] Remove redundant doc comments from Conv matchers (#172344)
-- This commit removes the redundant doc comments pertaining to indexing maps of convolution ops from their respective matchers. -- This is being done because the code self-documents the same and is a follow-up based on #172141 (comment) Signed-off-by: Abhishek Varma <[email protected]>
1 parent 1b4a74f commit 92abf7c

File tree

1 file changed

+0
-122
lines changed

1 file changed

+0
-122
lines changed

‎mlir/lib/Dialect/Linalg/Utils/Utils.cpp‎

Lines changed: 0 additions & 122 deletions
Original file line numberDiff line numberDiff line change
@@ -597,9 +597,6 @@ class ConvMatcherBuilder {
597597
// Matchers for specific convolution operation.
598598
//===----------------------------------------------------------------------===//
599599

600-
// #inputMap = affine_map<(W, w) -> (W + w)>
601-
// #filterMap = affine_map<(W, w) -> (w)>
602-
// #outputMap = affine_map<(W, w) -> (W)>
603600
template <>
604601
bool isaConvolutionOpOfType<linalg::Conv1DOp>(LinalgOp op,
605602
SmallVector<int64_t> *dilations,
@@ -621,9 +618,6 @@ bool isaConvolutionOpOfType<linalg::Conv1DOp>(LinalgOp op,
621618
.matchBody();
622619
}
623620

624-
// #inputMap = affine_map<(N, W, F, w, c) -> (N, W + w, c)>
625-
// #filterMap = affine_map<(N, W, F, w, c) -> (w, c, F)>
626-
// #outputMap = affine_map<(N, W, F, w, c) -> (N, W, F)>
627621
template <>
628622
bool isaConvolutionOpOfType<linalg::Conv1DNwcWcfOp>(
629623
LinalgOp op, SmallVector<int64_t> *dilations,
@@ -648,9 +642,6 @@ bool isaConvolutionOpOfType<linalg::Conv1DNwcWcfOp>(
648642
.matchBody();
649643
}
650644

651-
// #inputMap = affine_map<(N, F, W, c, w) -> (N, c, W + w)>
652-
// #filterMap = affine_map<(N, F, W, c, w) -> (F, c, w)>
653-
// #outputMap = affine_map<(N, F, W, c, w) -> (N, F, W)>
654645
template <>
655646
bool isaConvolutionOpOfType<linalg::Conv1DNcwFcwOp>(
656647
LinalgOp op, SmallVector<int64_t> *dilations,
@@ -675,9 +666,6 @@ bool isaConvolutionOpOfType<linalg::Conv1DNcwFcwOp>(
675666
.matchBody();
676667
}
677668

678-
// #inputMap = affine_map<(H, W, h, w) -> (H + h, W + w)>
679-
// #filterMap = affine_map<(H, W, h, w) -> (h, w)>
680-
// #outputMap = affine_map<(H, W, h, w) -> (H, W)>
681669
template <>
682670
bool isaConvolutionOpOfType<linalg::Conv2DOp>(LinalgOp op,
683671
SmallVector<int64_t> *dilations,
@@ -702,9 +690,6 @@ bool isaConvolutionOpOfType<linalg::Conv2DOp>(LinalgOp op,
702690
.matchBody();
703691
}
704692

705-
// #inputMap = affine_map<(N, H, W, F, h, w, c) -> (N, H + h, W + w, c)>
706-
// #filterMap = affine_map<(N, H, W, F, h, w, c) -> (h, w, c, F)>
707-
// #outputMap = affine_map<(N, H, W, F, h, w, c) -> (N, H, W, F)>
708693
template <>
709694
bool isaConvolutionOpOfType<linalg::Conv2DNhwcHwcfOp>(
710695
LinalgOp op, SmallVector<int64_t> *dilations,
@@ -732,10 +717,6 @@ bool isaConvolutionOpOfType<linalg::Conv2DNhwcHwcfOp>(
732717
.matchBody();
733718
}
734719

735-
// #inputMap = affine_map<(N, H, W, F, h, w, c) -> (N, H + h, W + w, c)>
736-
// #filterMap = affine_map<(N, H, W, F, h, w, c) -> (h, w, c, F)>
737-
// #scalarMap = affine_map<(N, H, W, F, h, w, c) -> ()>
738-
// #outputMap = affine_map<(N, H, W, F, h, w, c) -> (N, H, W, F)>
739720
template <>
740721
bool isaConvolutionOpOfType<linalg::Conv2DNhwcHwcfQOp>(
741722
LinalgOp op, SmallVector<int64_t> *dilations,
@@ -765,9 +746,6 @@ bool isaConvolutionOpOfType<linalg::Conv2DNhwcHwcfQOp>(
765746
.matchBody(/*containsZeroPointOffset=*/true);
766747
}
767748

768-
// #inputMap = affine_map<(N, H, W, F, h, w, c) -> (N, H + h, W + w, c)>
769-
// #filterMap = affine_map<(N, H, W, F, h, w, c) -> (F, h, w, c)>
770-
// #outputMap = affine_map<(N, H, W, F, h, w, c) -> (N, H, W, F)>
771749
template <>
772750
bool isaConvolutionOpOfType<linalg::Conv2DNhwcFhwcOp>(
773751
LinalgOp op, SmallVector<int64_t> *dilations,
@@ -795,10 +773,6 @@ bool isaConvolutionOpOfType<linalg::Conv2DNhwcFhwcOp>(
795773
.matchBody();
796774
}
797775

798-
// #inputMap = affine_map<(N, H, W, F, h, w, c) -> (N, H + h, W + w, c)>
799-
// #filterMap = affine_map<(N, H, W, F, h, w, c) -> (F, h, w, c)>
800-
// #scalarMap = affine_map<(N, H, W, F, h, w, c) -> ()>
801-
// #outputMap = affine_map<(N, H, W, F, h, w, c) -> (N, H, W, F)>
802776
template <>
803777
bool isaConvolutionOpOfType<linalg::Conv2DNhwcFhwcQOp>(
804778
LinalgOp op, SmallVector<int64_t> *dilations,
@@ -828,9 +802,6 @@ bool isaConvolutionOpOfType<linalg::Conv2DNhwcFhwcQOp>(
828802
.matchBody(/*containsZeroPointOffset=*/true);
829803
}
830804

831-
// #inputMap = affine_map<(N, F, H, W, c, h, w) -> (N, c, H + h, W + w)>
832-
// #filterMap = affine_map<(N, F, H, W, c, h, w) -> (F, c, h, w)>
833-
// #outputMap = affine_map<(N, F, H, W, c, h, w) -> (N, F, H, W)>
834805
template <>
835806
bool isaConvolutionOpOfType<linalg::Conv2DNchwFchwOp>(
836807
LinalgOp op, SmallVector<int64_t> *dilations,
@@ -858,10 +829,6 @@ bool isaConvolutionOpOfType<linalg::Conv2DNchwFchwOp>(
858829
.matchBody();
859830
}
860831

861-
// #inputMap = affine_map<(N, F, H, W, c, h, w) -> (N, c, H + h, W + w)>
862-
// #filterMap = affine_map<(N, F, H, W, c, h, w) -> (F, c, h, w)>
863-
// #scalarMap = affine_map<(N, F, H, W, c, h, w) -> ()>
864-
// #outputMap = affine_map<(N, F, H, W, c, h, w) -> (N, F, H, W)>
865832
template <>
866833
bool isaConvolutionOpOfType<linalg::Conv2DNchwFchwQOp>(
867834
LinalgOp op, SmallVector<int64_t> *dilations,
@@ -891,9 +858,6 @@ bool isaConvolutionOpOfType<linalg::Conv2DNchwFchwQOp>(
891858
.matchBody(/*containsZeroPointOffset=*/true);
892859
}
893860

894-
// #inputMap = affine_map<(N, G, F, H, W, c, h, w) -> (N, G, c, H + h, W + w)>
895-
// #filterMap = affine_map<(N, G, F, H, W, c, h, w) -> (F, G, c, h, w)>
896-
// #outputMap = affine_map<(N, G, F, H, W, c, h, w) -> (N, G, F, H, W)>
897861
template <>
898862
bool isaConvolutionOpOfType<linalg::Conv2DNgchwFgchwOp>(
899863
LinalgOp op, SmallVector<int64_t> *dilations,
@@ -923,9 +887,6 @@ bool isaConvolutionOpOfType<linalg::Conv2DNgchwFgchwOp>(
923887
.matchBody();
924888
}
925889

926-
// #inputMap = affine_map<(N, G, F, H, W, c, h, w) -> (N, G, c, H + h, W + w)>
927-
// #filterMap = affine_map<(N, G, F, H, W, c, h, w) -> (G, F, c, h, w)>
928-
// #outputMap = affine_map<(N, G, F, H, W, c, h, w) -> (N, G, F, H, W)>
929890
template <>
930891
bool isaConvolutionOpOfType<linalg::Conv2DNgchwGfchwOp>(
931892
LinalgOp op, SmallVector<int64_t> *dilations,
@@ -955,10 +916,6 @@ bool isaConvolutionOpOfType<linalg::Conv2DNgchwGfchwOp>(
955916
.matchBody();
956917
}
957918

958-
// #inputMap = affine_map<(N, G, F, H, W, c, h, w) -> (N, G, c, H + h, W + w)>
959-
// #filterMap = affine_map<(N, G, F, H, W, c, h, w) -> (G, F, c, h, w)>
960-
// #scalarMap = affine_map<(N, G, F, H, W, c, h, w) -> ()>
961-
// #outputMap = affine_map<(N, G, F, H, W, c, h, w) -> (N, G, F, H, W)>
962919
template <>
963920
bool isaConvolutionOpOfType<linalg::Conv2DNgchwGfchwQOp>(
964921
LinalgOp op, SmallVector<int64_t> *dilations,
@@ -990,9 +947,6 @@ bool isaConvolutionOpOfType<linalg::Conv2DNgchwGfchwQOp>(
990947
.matchBody(/*containsZeroPointOffset=*/true);
991948
}
992949

993-
// #inputMap = affine_map<(N, H, W, G, F, h, w, c) -> (N, H + h, W + w, G, c)>
994-
// #filterMap = affine_map<(N, H, W, G, F, h, w, c) -> (G, F, h, w, c)>
995-
// #outputMap = affine_map<(N, H, W, G, F, h, w, c) -> (N, H, W, G, F)>
996950
template <>
997951
bool isaConvolutionOpOfType<linalg::Conv2DNhwgcGfhwcOp>(
998952
LinalgOp op, SmallVector<int64_t> *dilations,
@@ -1022,10 +976,6 @@ bool isaConvolutionOpOfType<linalg::Conv2DNhwgcGfhwcOp>(
1022976
.matchBody();
1023977
}
1024978

1025-
// #inputMap = affine_map<(N, H, W, G, F, h, w, c) -> (N, H + h, W + w, G, c)>
1026-
// #filterMap = affine_map<(N, H, W, G, F, h, w, c) -> (G, F, h, w, c)>
1027-
// #scalarMap = affine_map<(N, H, W, G, F, h, w, c) -> ()>
1028-
// #outputMap = affine_map<(N, H, W, G, F, h, w, c) -> (N, H, W, G, F)>
1029979
template <>
1030980
bool isaConvolutionOpOfType<linalg::Conv2DNhwgcGfhwcQOp>(
1031981
LinalgOp op, SmallVector<int64_t> *dilations,
@@ -1057,9 +1007,6 @@ bool isaConvolutionOpOfType<linalg::Conv2DNhwgcGfhwcQOp>(
10571007
.matchBody(/*containsZeroPointOffset=*/true);
10581008
}
10591009

1060-
// #inputMap = affine_map<(D, H, W, d, h, w) -> (D + d, H + h, W + w)>
1061-
// #filterMap = affine_map<(D, H, W, d, h, w) -> (d, h, w)>
1062-
// #outputMap = affine_map<(D, H, W, d, h, w) -> (D, H, W)>
10631010
template <>
10641011
bool isaConvolutionOpOfType<linalg::Conv3DOp>(LinalgOp op,
10651012
SmallVector<int64_t> *dilations,
@@ -1088,10 +1035,6 @@ bool isaConvolutionOpOfType<linalg::Conv3DOp>(LinalgOp op,
10881035
.matchBody();
10891036
}
10901037

1091-
// #inputMap = affine_map<(N, D, H, W, F, d, h, w, c)
1092-
// -> (N, D + d, H + h, W + w, c)>
1093-
// #filterMap = affine_map<(N, D, H, W, F, d, h, w, c) -> (d, h, w, c, F)>
1094-
// #outputMap = affine_map<(N, D, H, W, F, d, h, w, c) -> (N, D, H, W, F)>
10951038
template <>
10961039
bool isaConvolutionOpOfType<linalg::Conv3DNdhwcDhwcfOp>(
10971040
LinalgOp op, SmallVector<int64_t> *dilations,
@@ -1123,11 +1066,6 @@ bool isaConvolutionOpOfType<linalg::Conv3DNdhwcDhwcfOp>(
11231066
.matchBody();
11241067
}
11251068

1126-
// #inputMap = affine_map<(N, D, H, W, F, d, h, w, c)
1127-
// -> (N, D + d, H + h, W + w, c)>
1128-
// #filterMap = affine_map<(N, D, H, W, F, d, h, w, c) -> (d, h, w, c, F)>
1129-
// #scalarMap = affine_map<(N, D, H, W, F, d, h, w, c) -> ()>
1130-
// #outputMap = affine_map<(N, D, H, W, F, d, h, w, c) -> (N, D, H, W, F)>
11311069
template <>
11321070
bool isaConvolutionOpOfType<linalg::Conv3DNdhwcDhwcfQOp>(
11331071
LinalgOp op, SmallVector<int64_t> *dilations,
@@ -1161,10 +1099,6 @@ bool isaConvolutionOpOfType<linalg::Conv3DNdhwcDhwcfQOp>(
11611099
.matchBody(/*containsZeroPointOffset=*/true);
11621100
}
11631101

1164-
// #inputMap = affine_map<(N, F, D, H, W, c, d, h, w)
1165-
// -> (N, c, D + d, H + h, W + w)>
1166-
// #filterMap = affine_map<(N, F, D, H, W, c, d, h, w) -> (F, c, d, h, w)>
1167-
// #outputMap = affine_map<(N, F, D, H, W, c, d, h, w) -> (N, F, D, H, W)>
11681102
template <>
11691103
bool isaConvolutionOpOfType<linalg::Conv3DNcdhwFcdhwOp>(
11701104
LinalgOp op, SmallVector<int64_t> *dilations,
@@ -1196,9 +1130,6 @@ bool isaConvolutionOpOfType<linalg::Conv3DNcdhwFcdhwOp>(
11961130
.matchBody();
11971131
}
11981132

1199-
// #inputMap = affine_map<(N, W, C, w) -> (N, C, W + w)>
1200-
// #filterMap = affine_map<(N, W, C, w) -> (C, w)>
1201-
// #outputMap = affine_map<(N, W, C, w) -> (N, C, W)>
12021133
template <>
12031134
bool isaConvolutionOpOfType<linalg::DepthwiseConv1DNcwCwOp>(
12041135
LinalgOp op, SmallVector<int64_t> *dilations,
@@ -1222,9 +1153,6 @@ bool isaConvolutionOpOfType<linalg::DepthwiseConv1DNcwCwOp>(
12221153
.matchBody();
12231154
}
12241155

1225-
// #inputMap = affine_map<(N, W, C, w) -> (N, W + w, C)>
1226-
// #filterMap = affine_map<(N, W, C, w) -> (w, C)>
1227-
// #outputMap = affine_map<(N, W, C, w) -> (N, W, C)>
12281156
template <>
12291157
bool isaConvolutionOpOfType<linalg::DepthwiseConv1DNwcWcOp>(
12301158
LinalgOp op, SmallVector<int64_t> *dilations,
@@ -1248,9 +1176,6 @@ bool isaConvolutionOpOfType<linalg::DepthwiseConv1DNwcWcOp>(
12481176
.matchBody();
12491177
}
12501178

1251-
// #inputMap = affine_map<(N, W, C, CM, w) -> (N, W + w, C)>
1252-
// #filterMap = affine_map<(N, W, C, CM, w) -> (w, C, CM)>
1253-
// #outputMap = affine_map<(N, W, C, CM, w) -> (N, W, C, CM)>
12541179
template <>
12551180
bool isaConvolutionOpOfType<linalg::DepthwiseConv1DNwcWcmOp>(
12561181
LinalgOp op, SmallVector<int64_t> *dilations,
@@ -1275,9 +1200,6 @@ bool isaConvolutionOpOfType<linalg::DepthwiseConv1DNwcWcmOp>(
12751200
.matchBody();
12761201
}
12771202

1278-
// #inputMap = affine_map<(N, H, W, C, h, w) -> (N, C, H + h, W + w)>
1279-
// #filterMap = affine_map<(N, H, W, C, h, w) -> (C, h, w)>
1280-
// #outputMap = affine_map<(N, H, W, C, h, w) -> (N, C, H, W)>
12811203
template <>
12821204
bool isaConvolutionOpOfType<linalg::DepthwiseConv2DNchwChwOp>(
12831205
LinalgOp op, SmallVector<int64_t> *dilations,
@@ -1304,9 +1226,6 @@ bool isaConvolutionOpOfType<linalg::DepthwiseConv2DNchwChwOp>(
13041226
.matchBody();
13051227
}
13061228

1307-
// #inputMap = affine_map<(N, H, W, C, h, w) -> (N, H + h, W + w, C)>
1308-
// #filterMap = affine_map<(N, H, W, C, h, w) -> (h, w, C)>
1309-
// #outputMap = affine_map<(N, H, W, C, h, w) -> (N, H, W, C)>
13101229
template <>
13111230
bool isaConvolutionOpOfType<linalg::DepthwiseConv2DNhwcHwcOp>(
13121231
LinalgOp op, SmallVector<int64_t> *dilations,
@@ -1333,10 +1252,6 @@ bool isaConvolutionOpOfType<linalg::DepthwiseConv2DNhwcHwcOp>(
13331252
.matchBody();
13341253
}
13351254

1336-
// #inputMap = affine_map<(N, H, W, C, h, w) -> (N, H + h, W + w, C)>
1337-
// #filterMap = affine_map<(N, H, W, C, h, w) -> (h, w, C)>
1338-
// #scalarMap = affine_map<(N, H, W, C, h, w) -> ()>
1339-
// #outputMap = affine_map<(N, H, W, C, h, w) -> (N, H, W, C)>
13401255
template <>
13411256
bool isaConvolutionOpOfType<linalg::DepthwiseConv2DNhwcHwcQOp>(
13421257
LinalgOp op, SmallVector<int64_t> *dilations,
@@ -1365,9 +1280,6 @@ bool isaConvolutionOpOfType<linalg::DepthwiseConv2DNhwcHwcQOp>(
13651280
.matchBody(/*containsZeroPointOffset=*/true);
13661281
}
13671282

1368-
// #inputMap = affine_map<(N, H, W, C, CM, h, w) -> (N, H + h, W + w, C)>
1369-
// #filterMap = affine_map<(N, H, W, C, CM, h, w) -> (h, w, C, CM)>
1370-
// #outputMap = affine_map<(N, H, W, C, CM, h, w) -> (N, H, W, C, CM)>
13711283
template <>
13721284
bool isaConvolutionOpOfType<linalg::DepthwiseConv2DNhwcHwcmOp>(
13731285
LinalgOp op, SmallVector<int64_t> *dilations,
@@ -1395,10 +1307,6 @@ bool isaConvolutionOpOfType<linalg::DepthwiseConv2DNhwcHwcmOp>(
13951307
.matchBody();
13961308
}
13971309

1398-
// #inputMap = affine_map<(N, H, W, C, CM, h, w) -> (N, H + h, W + w, C)>
1399-
// #filterMap = affine_map<(N, H, W, C, CM, h, w) -> (h, w, C, CM)>
1400-
// #scalarMap = affine_map<(N, H, W, C, CM, h, w) -> ()>
1401-
// #outputMap = affine_map<(N, H, W, C, CM, h, w) -> (N, H, W, C, CM)>
14021310
template <>
14031311
bool isaConvolutionOpOfType<linalg::DepthwiseConv2DNhwcHwcmQOp>(
14041312
LinalgOp op, SmallVector<int64_t> *dilations,
@@ -1428,12 +1336,6 @@ bool isaConvolutionOpOfType<linalg::DepthwiseConv2DNhwcHwcmQOp>(
14281336
.matchBody(/*containsZeroPointOffset=*/true);
14291337
}
14301338

1431-
// #inputMap = affine_map<(N, D, H, W, d, h, w, C)
1432-
// -> (N, D + d, H + h, W + w, C)>
1433-
// #filterMap = affine_map<(N, D, H, W, d, h, w, C)
1434-
// -> (d, h, w, C)>
1435-
// #outputMap = affine_map<(N, D, H, W, d, h, w, C)
1436-
// -> (N, D, H, W, C)>
14371339
template <>
14381340
bool isaConvolutionOpOfType<linalg::DepthwiseConv3DNdhwcDhwcOp>(
14391341
LinalgOp op, SmallVector<int64_t> *dilations,
@@ -1464,9 +1366,6 @@ bool isaConvolutionOpOfType<linalg::DepthwiseConv3DNdhwcDhwcOp>(
14641366
.matchBody();
14651367
}
14661368

1467-
// #inputMap = affine_map<(N, D, H, W, d, h, w, C) -> (N, C, D + d, H + h, W +
1468-
// w)> #filterMap = affine_map<(N, D, H, W, d, h, w, C) -> (C, d, h, w)>
1469-
// #outputMap = affine_map<(N, D, H, W, d, h, w, C) -> (N, C, D, H, W)>
14701369
template <>
14711370
bool isaConvolutionOpOfType<linalg::DepthwiseConv3DNcdhwCdhwOp>(
14721371
LinalgOp op, SmallVector<int64_t> *dilations,
@@ -1497,12 +1396,6 @@ bool isaConvolutionOpOfType<linalg::DepthwiseConv3DNcdhwCdhwOp>(
14971396
.matchBody();
14981397
}
14991398

1500-
// #inputMap = affine_map<(N, D, H, W, CM, d, h, w, C)
1501-
// -> (N, D + d, H + h, W + w, C)>
1502-
// #filterMap = affine_map<(N, D, H, W, CM, d, h, w, C)
1503-
// -> (d, h, w, C, CM)>
1504-
// #outputMap = affine_map<(N, D, H, W, CM, d, h, w, C)
1505-
// -> (N, D, H, W, C, CM)>
15061399
template <>
15071400
bool isaConvolutionOpOfType<linalg::DepthwiseConv3DNdhwcDhwcmOp>(
15081401
LinalgOp op, SmallVector<int64_t> *dilations,
@@ -1534,9 +1427,6 @@ bool isaConvolutionOpOfType<linalg::DepthwiseConv3DNdhwcDhwcmOp>(
15341427
.matchBody();
15351428
}
15361429

1537-
// #inputMap = affine_map<(N, H, W, C, h, w) -> (N, H + h, W + w, C)>
1538-
// #filterMap = affine_map<(N, H, W, C, h, w) -> (h, w)>
1539-
// #outputMap = affine_map<(N, H, W, C, h, w) -> (N, H, W, C)>
15401430
template <>
15411431
bool isaConvolutionOpOfType<linalg::PoolingNhwcMaxOp>(
15421432
LinalgOp op, SmallVector<int64_t> *dilations,
@@ -1564,9 +1454,6 @@ bool isaConvolutionOpOfType<linalg::PoolingNhwcMaxOp>(
15641454
.matchBody();
15651455
}
15661456

1567-
// #inputMap = affine_map<(N, H, W, C, h, w) -> (N, H + h, W + w, C)>
1568-
// #filterMap = affine_map<(N, H, W, C, h, w) -> (h, w)>
1569-
// #outputMap = affine_map<(N, H, W, C, h, w) -> (N, H, W, C)>
15701457
template <>
15711458
bool isaConvolutionOpOfType<linalg::PoolingNhwcMinOp>(
15721459
LinalgOp op, SmallVector<int64_t> *dilations,
@@ -1594,9 +1481,6 @@ bool isaConvolutionOpOfType<linalg::PoolingNhwcMinOp>(
15941481
.matchBody();
15951482
}
15961483

1597-
// #inputMap = affine_map<(N, H, W, C, h, w) -> (N, H + h, W + w, C)>
1598-
// #filterMap = affine_map<(N, H, W, C, h, w) -> (h, w)>
1599-
// #outputMap = affine_map<(N, H, W, C, h, w) -> (N, H, W, C)>
16001484
template <>
16011485
bool isaConvolutionOpOfType<linalg::PoolingNhwcSumOp>(
16021486
LinalgOp op, SmallVector<int64_t> *dilations,
@@ -1624,9 +1508,6 @@ bool isaConvolutionOpOfType<linalg::PoolingNhwcSumOp>(
16241508
.matchBody();
16251509
}
16261510

1627-
// #inputMap = affine_map<(N, H, W, C, h, w) -> (N, H + h, W + w, C)>
1628-
// #filterMap = affine_map<(N, H, W, C, h, w) -> (h, w)>
1629-
// #outputMap = affine_map<(N, H, W, C, h, w) -> (N, H, W, C)>
16301511
template <>
16311512
bool isaConvolutionOpOfType<linalg::PoolingNhwcMaxUnsignedOp>(
16321513
LinalgOp op, SmallVector<int64_t> *dilations,
@@ -1654,9 +1535,6 @@ bool isaConvolutionOpOfType<linalg::PoolingNhwcMaxUnsignedOp>(
16541535
.matchBody();
16551536
}
16561537

1657-
// #inputMap = affine_map<(N, H, W, C, h, w) -> (N, H + h, W + w, C)>
1658-
// #filterMap = affine_map<(N, H, W, C, h, w) -> (h, w)>
1659-
// #outputMap = affine_map<(N, H, W, C, h, w) -> (N, H, W, C)>
16601538
template <>
16611539
bool isaConvolutionOpOfType<linalg::PoolingNhwcMinUnsignedOp>(
16621540
LinalgOp op, SmallVector<int64_t> *dilations,

0 commit comments

Comments
 (0)