Home
last modified time | relevance | path

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

/external/pytorch/torch/_dynamo/variables/
Dctx_manager.py76 self.set_cleanup_hook(tx)
79 def set_cleanup_hook(self, tx: "InstructionTranslator", fn=None): member in ContextWrappingVariable
202 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 …]