Method gather

lib/nn/parallel/data_parallel.py:49–50 · view source on GitHub ↗

(self, outputs, output_device)

Source from the content-addressed store, hash-verified

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

dict_gatherFunction · 0.85

no test coverage detected