Searched refs:_has_fsdp_params (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/distributed/fsdp/ |
| H A D | _state_dict_utils.py | 32 _has_fsdp_params,
87 if not _has_fsdp_params(fsdp_state, module):
194 if not state_dict or not _has_fsdp_params(fsdp_state, module):
392 _has_fsdp_params(fsdp_state, module)
416 if not _has_fsdp_params(fsdp_state, module):
473 assert not _has_fsdp_params(fsdp_state, module), (
517 _has_fsdp_params(fsdp_state, module)
578 if _has_fsdp_params(fsdp_state, module):
597 if not _has_fsdp_params(fsdp_state, module):
|
| H A D | _unshard_param_utils.py | 12 _has_fsdp_params,
82 if _has_fsdp_params(state, module):
97 if _has_fsdp_params(state, module):
|
| H A D | _common_utils.py | 226 def _has_fsdp_params(state: _FSDPState, module: nn.Module) -> bool: function
|