Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/distributed/_composable/fsdp/
H A Dfully_shard.py302 _assert_all_fsdp_modules(modules)
322 _assert_all_fsdp_modules(modules)
443 def _assert_all_fsdp_modules(modules: Iterable[Any]) -> None: function