Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Ddistributed.py313 user_hooks: VariableTracker,
350 user_hooks_name, _ = tx.output.add_backward_state_hook(user_hooks)
358 return BackwardHookVariable(proxy, module, user_hooks, user_pre_hooks)
364 user_hooks: VariableTracker,
371 self.user_hooks = user_hooks
/aosp_15_r20/external/pytorch/torch/utils/
H A Dhooks.py104 def __init__(self, module, user_hooks, user_pre_hooks): argument
105 self.user_hooks = user_hooks
138 for hook in self.user_hooks:
229 for user_hook in self.user_hooks: