31   GpuAllReduceRewriter(gpu::GPUFuncOp funcOp, gpu::AllReduceOp reduceOp,  in GpuAllReduceRewriter()
33       : funcOp(funcOp), reduceOp(reduceOp), rewriter(rewriter),  in GpuAllReduceRewriter()
34         loc(reduceOp.getLoc()), valueType(reduceOp.value().getType()),  in GpuAllReduceRewriter()
35         indexType(IndexType::get(reduceOp.getContext())),  in GpuAllReduceRewriter()
36         int32Type(IntegerType::get(reduceOp.getContext(), /*width=*/32)) {}  in GpuAllReduceRewriter()
66     rewriter.setInsertionPoint(reduceOp);  in rewrite()
104                                                 reduceOp.value(), accumFactory);  in rewrite()
141     rewriter.replaceOp(reduceOp, result);  in rewrite()
171     auto &body = reduceOp.body();  in getFactory()
174     auto opAttr = reduceOp.op();  in getFactory()
379   gpu::AllReduceOp reduceOp;  member
397     auto callback = [&](gpu::AllReduceOp reduceOp) {  in matchAndRewrite()  argument
398       GpuAllReduceRewriter(funcOp, reduceOp, rewriter).rewrite();  in matchAndRewrite()