Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/distributed/fsdp/
H A Dtest_fsdp_meta.py58 class MyLinear(nn.Linear): class
/aosp_15_r20/external/pytorch/test/higher_order_ops/
H A Dtest_with_effects.py351 class MyLinear(torch.nn.Module): class
/aosp_15_r20/external/pytorch/test/export/
H A Dtest_export.py1460 class MyLinear(torch.nn.Module): class
1545 class MyLinear(torch.nn.Module): class
2870 class MyLinear(torch.nn.Module): class
3649 class MyLinear(torch.nn.Module): class
3706 class MyLinear(torch.nn.Module): class
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_export.py3961 class MyLinear(torch.nn.Module): class
3998 class MyLinear(torch.nn.Module): class
4048 class MyLinear(torch.nn.Module): class
/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_dynamo_distributed.py203 class MyLinear(torch.nn.Module): class