Searched defs:DataParallel (Results 1 – 4 of 4) sorted by relevance
844 .. _cudaGetDeviceCount:847 .. _cuInit:850 .. _nvmlDeviceGetCount_v2:853 .. _cuda-graph-semantics:
19 model: Union[torch.nn.Module, torch.nn.DataParallel],
18 from torch.nn.parallel import DataParallel as DataParallel unknown
52 class DataParallel(Module, Generic[T]): class