Home
last modified time | relevance | path

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

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