Searched refs:GetNcclAllReduceConfig (Results 1 – 1 of 1) sorted by relevance
89 static NcclAllReduceConfig GetNcclAllReduceConfig(mlir::lmhlo::AllReduceOp op, in GetNcclAllReduceConfig() function114 config_(GetNcclAllReduceConfig(op, replica_count)), in NcclAllReduceThunk()