Searched defs:GpuAllReduceRewrite (Results 1 – 1 of 1) sorted by relevance
350 struct GpuAllReduceRewrite : public RewritePattern { struct351 explicit GpuAllReduceRewrite(MLIRContext *context) in GpuAllReduceRewrite() argument354 LogicalResult matchAndRewrite(Operation *op, in matchAndRewrite()