Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DThreadLocalState.h88 friend class ThreadLocalStateGuard; variable
92 class TORCH_API ThreadLocalStateGuard {
94 explicit ThreadLocalStateGuard(const ThreadLocalState& state) in ThreadLocalStateGuard() function
100 ~ThreadLocalStateGuard() { in ~ThreadLocalStateGuard()
114 ThreadLocalStateGuard g(tls_state); in wrapPropagateTLSState()
H A DParallelThreadPoolNative.cpp86 ThreadLocalStateGuard guard(thread_locals); in launch()
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dengine.cpp558 at::ThreadLocalStateGuard tls_guard(local_graph_task->thread_locals_); in thread_main()
772 at::ThreadLocalStateGuard tls_guard(this->thread_locals_); in exec_post_processing()
/aosp_15_r20/external/pytorch/torch/csrc/distributed/autograd/engine/
H A Ddist_engine.cpp375 at::ThreadLocalStateGuard tls_guard(local_graph_task->thread_locals_); in execute_graph_task_until_ready_queue_empty()
/aosp_15_r20/external/pytorch/torch/csrc/dynamo/
H A Dpython_compiled_autograd.cpp717 at::ThreadLocalStateGuard tls_guard(graph_task.thread_locals_); in compiled_autograd()
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dinterpreter.cpp1247 at::ThreadLocalStateGuard g(*tls_state_); in operator ()()
/aosp_15_r20/external/pytorch/test/cpp/jit/
H A Dtest_misc.cpp1258 ThreadLocalStateGuard g_tls(state); in TEST()