(self, *inputs, **kwargs)
| 89 | self.allreduce_params = allreduce_params |
| 90 | |
| 91 | def forward(self, *inputs, **kwargs): |
| 92 | self.needs_reduction = True |
| 93 | return self.module(*inputs, **kwargs) |
| 94 | |
| 95 | def state_dict(self, destination=None, prefix='', keep_vars=False): |
| 96 | # [h.remove() for h in self.hook_handles] |
nothing calls this directly
no outgoing calls
no test coverage detected