Searched refs:reduceOp (Results 1 – 2 of 2) sorted by relevance
31 : funcOp(funcOp_), reduceOp(reduceOp_), rewriter(rewriter_), in GpuAllReduceRewriter()32 loc(reduceOp.getLoc()), valueType(reduceOp.value().getType()), in GpuAllReduceRewriter()33 indexType(IndexType::get(reduceOp.getContext())), in GpuAllReduceRewriter()34 int32Type(IntegerType::get(/*width=*/32, reduceOp.getContext())) {} in GpuAllReduceRewriter()64 rewriter.setInsertionPoint(reduceOp); in rewrite()100 reduceOp.value(), accumFactory); in rewrite()137 rewriter.replaceOp(reduceOp, result); in rewrite()165 auto &body = reduceOp.body(); in getFactory()168 auto opAttr = reduceOp.op(); in getFactory()369 gpu::AllReduceOp reduceOp; member[all …]
907 auto reduceOp = std::get<1>(resultAndReduce); in verify() local908 auto reduceType = reduceOp.operand().getType(); in verify()910 return reduceOp.emitOpError() in verify()1175 auto reduceOp = cast<ReduceOp>(op->getParentOp()); in verify() local1176 Type reduceType = reduceOp.operand().getType(); in verify()