Home
last modified time | relevance | path

Searched defs:MLPModule (Results 1 – 5 of 5) sorted by relevance

/external/pytorch/test/distributed/pipelining/
Dtest_transformer.py13 class MLPModule(torch.nn.Module): class
Dmodel_registry.py77 class MLPModule(torch.nn.Module): class
/external/pytorch/test/distributed/_composable/test_composability/
Dtest_pp_composability.py39 class MLPModule(torch.nn.Module): class
/external/pytorch/torch/testing/_internal/distributed/_tensor/
Dcommon_dtensor.py64 class MLPModule(nn.Module): class
/external/pytorch/test/dynamo/
Dtest_activation_checkpointing.py1048 class MLPModule(nn.Module): class