Searched refs:param_to_hook_all_reduce (Results 1 – 1 of 1) sorted by relevance
642 param_to_hook_all_reduce=None, argument650 param_to_hook_all_reduce is not None816 param_to_hook_all_reduce=param_to_hook_all_reduce,969 param_to_hook_all_reduce, argument984 param_to_hook_all_reduce.register_hook(self._delayed_all_reduce_hook)