Searched refs:run_allreduce (Results 1 – 1 of 1) sorted by relevance
145 def run_allreduce(op): function163 run_allreduce(ReduceOp.SUM)164 run_allreduce(ReduceOp.MAX)165 run_allreduce(ReduceOp.MIN)166 run_allreduce(ReduceOp.PROD)