Home
last modified time | relevance | path

Searched defs:_get_module_fsdp_state (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/_composable/fsdp/
H A D_fsdp_state.py334 def _get_module_fsdp_state(module: nn.Module) -> Optional[FSDPState]: function
/aosp_15_r20/external/pytorch/torch/distributed/fsdp/
H A D_common_utils.py158 def _get_module_fsdp_state(module: nn.Module) -> Optional[_FSDPState]: function