Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dmutation_guard.py11 from .utils import ExactWeakKeyDictionary, is_lazy_module, nn_module_has_global_hooks
117 if isinstance(obj, torch.nn.Module) and nn_module_has_global_hooks():
H A Dutils.py2111 def nn_module_has_global_hooks(): function