Searched refs:allreduce_with_then_hook (Results 1 – 2 of 2) sorted by relevance
2019 def allreduce_with_then_hook( function2037 process_group, allreduce_with_then_hook
5409 def allreduce_with_then_hook( function5425 num_iters=4, ddp_comm_hook=allreduce_with_then_hook