Searched refs:_delayed_all_reduce_hook (Results 1 – 1 of 1) sorted by relevance
957 def _delayed_all_reduce_hook(self, grad): member in DistributedDataParallel984 param_to_hook_all_reduce.register_hook(self._delayed_all_reduce_hook)