AlwaysWrapNestedWrappedModule (reference) in projects: aosp_15_r20

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

/aosp_15_r20/external/pytorch/test/distributed/fsdp/
H A D	test_fsdp_core.py	`24 AlwaysWrapNestedWrappedModule, 155 AlwaysWrapNestedWrappedModule,`
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A D	common_fsdp.py	`478 class AlwaysWrapNestedWrappedModule(NestedWrappedModule): class 494 AlwaysWrapNestedWrappedModule, AlwaysWrapNestedWrappedModule`