Lines Matching refs:reduceOp
177 static Operation *buildMultiDimReduce(OpBuilder &b, Operation *reduceOp, in buildMultiDimReduce() argument
180 auto maybeKind = getCombinerOpKind(reduceOp); in buildMultiDimReduce()
183 reduceOp->getLoc(), valueToReduce, acc, reductionMask, *maybeKind); in buildMultiDimReduce()
381 Operation *reduceOp = in vectorizeOneOp() local
384 if (reduceOp) in vectorizeOneOp()
385 return VectorizationResult{VectorizationStatus::NewOp, reduceOp}; in vectorizeOneOp()
542 Operation *reduceOp = matchLinalgReduction(opOperand); in reductionPreconditions() local
543 if (!reduceOp || !getCombinerOpKind(reduceOp)) { in reductionPreconditions()
1338 Operation *reduceOp = matchLinalgReduction(linalgOp.getOutputOperand(0)); in Conv1DNwcGenerator() local
1339 if (!reduceOp) in Conv1DNwcGenerator()
1342 maybeKind = getCombinerOpKind(reduceOp); in Conv1DNwcGenerator()
1348 for (Value operand : reduceOp->getOperands()) { in Conv1DNwcGenerator()