Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/c10/core/impl/
H A DPythonDispatcherTLS.h8 struct C10_API PythonDispatcherTLS { struct
15 DisablePythonDispatcher() : old_(PythonDispatcherTLS::get_state()) { in DisablePythonDispatcher()
16 PythonDispatcherTLS::set_state({}); in DisablePythonDispatcher()
19 PythonDispatcherTLS::set_state(old_); in ~DisablePythonDispatcher()
H A DPythonDispatcherTLS.cpp9 void PythonDispatcherTLS::set_state(PyInterpreter* state) { in set_state()
14 PythonDispatcherTLS::reset_state(); in set_state()
19 PyInterpreter* PythonDispatcherTLS::get_state() { in get_state()
23 void PythonDispatcherTLS::reset_state() { in reset_state()
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DThreadLocalState.cpp19 …DispatchModeTLS::get_state()), python_dispatcher_state_(c10::impl::PythonDispatcherTLS::get_state(… in ThreadLocalState()
53 c10::impl::PythonDispatcherTLS::set_state(state.python_dispatcher_state_); in setThreadLocalState()
/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A DPythonFallbackKernel.cpp97 auto* state = c10::impl::PythonDispatcherTLS::get_state(); in pythonDispatcherFallback()
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dinit.cpp73 EnablePythonDispatcher() : old_(c10::impl::PythonDispatcherTLS::get_state()) { in EnablePythonDispatcher()
74 c10::impl::PythonDispatcherTLS::set_state(getPyInterpreter()); in EnablePythonDispatcher()
77 c10::impl::PythonDispatcherTLS::set_state(old_); in ~EnablePythonDispatcher()