Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/c10/core/impl/
H A DTorchDispatchModeTLS.cpp30 c10::impl::tls_set_dispatch_key_included(DispatchKey::Python, true); in push_non_infra_mode_onto_stack()
31 c10::impl::tls_set_dispatch_key_included( in push_non_infra_mode_onto_stack()
58 c10::impl::tls_set_dispatch_key_included(DispatchKey::Python, false); in pop_stack()
59 c10::impl::tls_set_dispatch_key_included( in pop_stack()
75 c10::impl::tls_set_dispatch_key_included(DispatchKey::Python, false); in pop_highest_infra_mode()
76 c10::impl::tls_set_dispatch_key_included( in pop_highest_infra_mode()
140 c10::impl::tls_set_dispatch_key_included(DispatchKey::Python, true); in set_mode()
141 c10::impl::tls_set_dispatch_key_included( in set_mode()
154 c10::impl::tls_set_dispatch_key_included(DispatchKey::Python, false); in unset_mode()
155 c10::impl::tls_set_dispatch_key_included( in unset_mode()
[all …]
H A DPythonDispatcherTLS.cpp11 c10::impl::tls_set_dispatch_key_included( in set_state()
25 c10::impl::tls_set_dispatch_key_included( in reset_state()
H A DLocalDispatchKeySet.cpp98 void tls_set_dispatch_key_included(DispatchKey x, bool desired_state) { in tls_set_dispatch_key_included() function
H A DLocalDispatchKeySet.h160 C10_API void tls_set_dispatch_key_included(DispatchKey x, bool desired_state);
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DLegacyVmapMode.cpp14 c10::impl::tls_set_dispatch_key_included(DispatchKey::VmapMode, true); in increment_nesting()
22 c10::impl::tls_set_dispatch_key_included(DispatchKey::VmapMode, false); in decrement_nesting()
H A DTracerMode.h125 c10::impl::tls_set_dispatch_key_included(at::DispatchKey::Tracer, enabled); in set_dispatch_enabled()
/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dpython_dispatch.cpp87 c10::impl::tls_set_dispatch_key_included(at::DispatchKey::Python, false); in EnableHermeticPyObject()
88 c10::impl::tls_set_dispatch_key_included( in EnableHermeticPyObject()
95 c10::impl::tls_set_dispatch_key_included( in ~EnableHermeticPyObject()
97 c10::impl::tls_set_dispatch_key_included( in ~EnableHermeticPyObject()
660 c10::impl::tls_set_dispatch_key_included(dispatch_key, desired_state); in initDispatchBindings()
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DDynamicLayer.cpp23 c10::impl::tls_set_dispatch_key_included(DispatchKey::FuncTorchDynamicLayerFrontMode, included); in setDynamicLayerFrontBackKeysIncluded()
24 c10::impl::tls_set_dispatch_key_included(DispatchKey::FuncTorchDynamicLayerBackMode, included); in setDynamicLayerFrontBackKeysIncluded()
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dpython_torch_functions_manual.cpp769 c10::impl::tls_set_dispatch_key_included( in initTorchFunctions()
778 c10::impl::tls_set_dispatch_key_included( in initTorchFunctions()