Searched refs:setup_stacktrace_preservation_hooks (Results 1 – 5 of 5) sorted by relevance
73 def setup_stacktrace_preservation_hooks(roots: List): function
49 from .logging_utils import setup_stacktrace_preservation_hooks231 setup_stacktrace_preservation_hooks([out.grad_fn for out in needed_outs])
229 setup_stacktrace_preservation_hooks,232 setup_stacktrace_preservation_hooks([loss.grad_fn])
918 from torch._functorch.aot_autograd import setup_stacktrace_preservation_hooks929 setup_stacktrace_preservation_hooks([out.grad_fn for out in outs])
75 setup_stacktrace_preservation_hooks,