Home
last modified time | relevance | path

Searched refs:tls_is_dispatch_key_excluded (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DTracerMode.h118 !c10::impl::tls_is_dispatch_key_excluded(at::DispatchKey::Tracer); in is_dispatch_enabled()
123 !c10::impl::tls_is_dispatch_key_excluded(at::DispatchKey::Tracer), in set_dispatch_enabled()
H A DSparseCsrTensorImpl.h118 !c10::impl::tls_is_dispatch_key_excluded(DispatchKey::Python)) { in shallow_copy_and_detach_core()
124 !c10::impl::tls_is_dispatch_key_excluded(DispatchKey::Python)) { in shallow_copy_and_detach_core()
H A DSparseTensorImpl.h317 !c10::impl::tls_is_dispatch_key_excluded(DispatchKey::Python)) { in shallow_copy_and_detach_core()
323 !c10::impl::tls_is_dispatch_key_excluded(DispatchKey::Python)) { in shallow_copy_and_detach_core()
H A DNestedTensorImpl.cpp302 !c10::impl::tls_is_dispatch_key_excluded(DispatchKey::Python)) { in shallow_copy_and_detach_core()
H A Dautocast_mode.cpp11 return !c10::impl::tls_is_dispatch_key_excluded(dispatch_key); in is_autocast_enabled()
H A DFunctionalTensorWrapper.cpp443 !c10::impl::tls_is_dispatch_key_excluded(DispatchKey::Python)) { in shallow_copy_and_detach_core()
/aosp_15_r20/external/pytorch/c10/core/impl/
H A DLocalDispatchKeySet.cpp78 bool tls_is_dispatch_key_excluded(DispatchKey x) { in tls_is_dispatch_key_excluded() function
H A DLocalDispatchKeySet.h157 C10_API bool tls_is_dispatch_key_excluded(DispatchKey x);
H A DTorchDispatchModeTLS.cpp181 return !c10::impl::tls_is_dispatch_key_excluded(DispatchKey::Python) && in dispatch_mode_enabled()
/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dpython_dispatch.cpp80 c10::impl::tls_is_dispatch_key_excluded(at::DispatchKey::Python)), in EnableHermeticPyObject()
244 : k(k), old(c10::impl::tls_is_dispatch_key_excluded(k)) { in SetExcludeDispatchKeyGuard()
655 return c10::impl::tls_is_dispatch_key_excluded(dispatch_key); in initDispatchBindings()
/aosp_15_r20/external/pytorch/c10/core/
H A DTensorImpl.cpp502 !c10::impl::tls_is_dispatch_key_excluded(DispatchKey::Python)) { in shallow_copy_and_detach_core()
508 !c10::impl::tls_is_dispatch_key_excluded(DispatchKey::Python)) { in shallow_copy_and_detach_core()
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dinference_mode.cpp32 ASSERT_FALSE(c10::impl::tls_is_dispatch_key_excluded( in assert_TLS_states()