Searched defs:allreduce_hook (Results 1 – 6 of 6) sorted by relevance
7 def allreduce_hook(state, bucket): function
33 def allreduce_hook( function
95 def allreduce_hook(state: DefaultState, grad: torch.Tensor): function
944 _AllReduceBySumCommHook allreduce_hook(process_group_); in run_allreduce_hook() local
1793 def allreduce_hook( function
5380 def allreduce_hook( function