Lines Matching defs:reductionOp
1694 auto reductionOp =
1696 auto vectorType = cast<VectorType>(reductionOp.getVector().getType());
1705 if (!reductionOp.getType().isIntOrFloat())
1713 SmallVector<Value> yieldValues = {reductionOp.getVector()};
1715 VectorType::get({numElements}, reductionOp.getType())};
1716 if (reductionOp.getAcc()) {
1717 yieldValues.push_back(reductionOp.getAcc());
1718 retTypes.push_back(reductionOp.getAcc().getType());
1729 distributedReductionFn(reductionOp.getLoc(), rewriter, laneValVec,
1730 reductionOp.getKind(), newWarpOp.getWarpSize());
1731 if (reductionOp.getAcc()) {
1733 rewriter, reductionOp.getLoc(), reductionOp.getKind(), fullReduce,