We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 13f5183 commit 2b5d1dcCopy full SHA for 2b5d1dc
mlir/lib/Dialect/Linalg/Transforms/Vectorization.cpp
@@ -1948,7 +1948,10 @@ static bool isCastOfBlockArgument(Operation *op) {
1948
isa<BlockArgument>(op->getOperand(0));
1949
}
1950
1951
-// Returns true iff it is a valid conv/pooling op.
+// Returns the ConvOperationKind of the op using reduceOp of the generic
1952
+// payload. If it is neither a convolution nor a pooling, it returns
1953
+// std::nullopt.
1954
+//
1955
// If (region has 2 ops (reduction + yield) or 3 ops (extension + reduction
1956
// + yield) and rhs is not used) then it is the body of a pooling
1957
// If conv, check for single `mul` predecessor. The `mul` operands must be
0 commit comments