Searched refs:user_hooks (Results 1 – 2 of 2) sorted by relevance
313 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
104 def __init__(self, module, user_hooks, user_pre_hooks): argument105 self.user_hooks = user_hooks138 for hook in self.user_hooks:229 for user_hook in self.user_hooks: