(self, module, device_ids)
| 190 | return outputs |
| 191 | |
| 192 | def replicate(self, module, device_ids): |
| 193 | return replicate(module, device_ids) |
| 194 | |
| 195 | def scatter(self, inputs, kwargs, device_ids): |
| 196 | return scatter_kwargs(inputs, kwargs, device_ids, dim=self.dim) |