Searched refs:set_cleanup_hook (Results 1 – 1 of 1) sorted by relevance
76 self.set_cleanup_hook(tx)79 def set_cleanup_hook(self, tx: "InstructionTranslator", fn=None): member in ContextWrappingVariable202 self.set_cleanup_hook(249 self.set_cleanup_hook(283 self.set_cleanup_hook(322 self.set_cleanup_hook(366 self.set_cleanup_hook(tx, lambda: torch._C._functorch._grad_decrement_nesting())404 self.set_cleanup_hook(tx, lambda: ctx_val.__exit__(None, None, None))437 self.set_cleanup_hook(tx, lambda: torch._C._functorch._vmap_decrement_nesting())545 self.set_cleanup_hook([all …]