Searched refs:reduceBlock (Results 1 – 2 of 2) sorted by relevance
494 Block &reduceBlock = reduce.getReductionOperator().front(); in matchAndRewrite() local496 yieldOperands.push_back(reduceBlock.getTerminator()->getOperand(0)); in matchAndRewrite()497 rewriter.eraseOp(reduceBlock.getTerminator()); in matchAndRewrite()498 rewriter.mergeBlockBefore(&reduceBlock, &op, {arg, reduce.getOperand()}); in matchAndRewrite()
2357 Block &reduceBlock = reduce.getReductionOperator().front(); in matchAndRewrite() local2359 mapping.map(reduceBlock.getArgument(0), op.getInitVals()[initValIndex]); in matchAndRewrite()2360 mapping.map(reduceBlock.getArgument(1), in matchAndRewrite()2362 for (auto &reduceBodyOp : reduceBlock.without_terminator()) in matchAndRewrite()2366 cast<ReduceReturnOp>(reduceBlock.getTerminator()).getResult()); in matchAndRewrite()