Searched refs:user_pre_hooks (Results 1 – 2 of 2) sorted by relevance
314 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
104 def __init__(self, module, user_hooks, user_pre_hooks): argument106 self.user_pre_hooks = user_pre_hooks209 if self.user_pre_hooks:211 for user_pre_hook in self.user_pre_hooks: