Searched refs:custom_reduce_scatter (Results 1 – 1 of 1) sorted by relevance
90 def custom_reduce_scatter(self, output, input, group=None): member in DummyHook108 self.custom_reduce_scatter(output, grad, group=state.process_group)