Home
last modified time | relevance | path

Searched refs:ModelParallelModel (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/test/distributed/optim/
Dtest_zero_redundancy_optimizer.py1142 class ModelParallelModel(torch.nn.Module): class
1169 mp_model = ModelParallelModel(dev0, dev1)