MCPcopy Create free account
hub / github.com/pytorch/pytorch / _apply

Method _apply

torch/distributed/tensor/parallel/style.py:28–29  ·  view source on GitHub ↗
(self, module: nn.Module, device_mesh: DeviceMesh)

Source from the content-addressed store, hash-verified

26
27 @abstractmethod
28 def _apply(self, module: nn.Module, device_mesh: DeviceMesh) -> nn.Module:
29 ...
30
31
32class ColwiseParallel(ParallelStyle):

Callers 1

parallelize_moduleFunction · 0.45

Calls

no outgoing calls

Tested by

no test coverage detected