Searched refs:_insert_module_state (Results 1 – 3 of 3) sorted by relevance
9 from torch.distributed._composable_state import _get_module_state, _insert_module_state124 _insert_module_state(module, state)130 _insert_module_state(submodule, state)
13 def _insert_module_state(module: nn.Module, state: _State) -> None: function
25 _insert_module_state,93 _insert_module_state(module, self)