Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A DPythonFallbackKernel.h7 struct TORCH_API RestorePythonTLSSnapshot { struct
8 RestorePythonTLSSnapshot();
9 ~RestorePythonTLSSnapshot();
H A DPythonFallbackKernel.cpp126 RestorePythonTLSSnapshot::RestorePythonTLSSnapshot() : saved_(safe_get_tls_on_entry()), guard_(safe… in RestorePythonTLSSnapshot() function in at::impl::RestorePythonTLSSnapshot
130 RestorePythonTLSSnapshot::~RestorePythonTLSSnapshot() { in ~RestorePythonTLSSnapshot()
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dinit.cpp498 py_context_manager<at::impl::RestorePythonTLSSnapshot>( in THPAutograd_initExtension()