Searched refs:_module_state_mapping (Results 1 – 1 of 1) sorted by relevance
10 _module_state_mapping: Dict[nn.Module, _State] = {} variable14 global _module_state_mapping15 assert module not in _module_state_mapping, f"Inserting {module} more than once."16 _module_state_mapping[module] = state29 global _module_state_mapping34 if module in _module_state_mapping:35 return _module_state_mapping[module]