| 46 | |
| 47 | |
| 48 | class DictGatherDataParallel(nn.DataParallel): |
| 49 | def gather(self, outputs, output_device): |
| 50 | return dict_gather(outputs, output_device, dim=self.dim) |
| 51 | |
| 52 | |
| 53 | class UserScatteredDataParallel(DictGatherDataParallel): |
nothing calls this directly
no outgoing calls
no test coverage detected