Lines Matching defs:reduceOp
608 static Operation *buildMultiDimReduce(OpBuilder &b, Operation *reduceOp,
611 auto maybeKind = getCombinerOpKind(reduceOp);
614 reduceOp->getLoc(), valueToReduce, acc, dimsToMask, *maybeKind);
1284 Operation *reduceOp =
1287 if (reduceOp)
1288 return VectorizationResult{VectorizationStatus::NewOp, reduceOp};
1798 Operation *reduceOp = matchLinalgReduction(&opOperand);
1799 if (!reduceOp || !getCombinerOpKind(reduceOp)) {
3074 Operation *reduceOp = matchLinalgReduction(linalgOp.getDpsInitOperand(0));
3075 if (!reduceOp)
3077 redOp = reduceOp->getName().getIdentifier();
3079 if (!setOperKind(reduceOp))
3081 auto maybeKind = getCombinerOpKind(reduceOp);
3777 bool setOperKind(Operation *reduceOp) {
3779 llvm::count_if(reduceOp->getOperands(), llvm::IsaPred<BlockArgument>);
3786 auto feedValIt = llvm::find_if_not(reduceOp->getOperands(),