Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Ddistributed.py314 user_pre_hooks: VariableTracker,
349 user_pre_hooks_name, _ = tx.output.add_backward_state_hook(user_pre_hooks)
358 return BackwardHookVariable(proxy, module, user_hooks, user_pre_hooks)
365 user_pre_hooks: VariableTracker,
372 self.user_pre_hooks = user_pre_hooks
/aosp_15_r20/external/pytorch/torch/utils/
H A Dhooks.py104 def __init__(self, module, user_hooks, user_pre_hooks): argument
106 self.user_pre_hooks = user_pre_hooks
209 if self.user_pre_hooks:
211 for user_pre_hook in self.user_pre_hooks: