Searched refs:tls_set_dispatch_key_excluded (Results 1 – 6 of 6) sorted by relevance
86 c10::impl::tls_set_dispatch_key_excluded(at::DispatchKey::Python, true); in EnableHermeticPyObject()93 c10::impl::tls_set_dispatch_key_excluded( in ~EnableHermeticPyObject()245 c10::impl::tls_set_dispatch_key_excluded(k, set_excluded); in SetExcludeDispatchKeyGuard()248 c10::impl::tls_set_dispatch_key_excluded(k, old); in ~SetExcludeDispatchKeyGuard()650 c10::impl::tls_set_dispatch_key_excluded(dispatch_key, desired_state); in initDispatchBindings()
82 void tls_set_dispatch_key_excluded(DispatchKey x, bool desired_state) { in tls_set_dispatch_key_excluded() function
158 C10_API void tls_set_dispatch_key_excluded(DispatchKey x, bool desired_state);
16 c10::impl::tls_set_dispatch_key_excluded(DispatchKey::Named, !enabled); in set_enabled()
16 c10::impl::tls_set_dispatch_key_excluded(dispatch_key, !enabled); in set_autocast_enabled()
386 c10::impl::tls_set_dispatch_key_excluded( in tls_set_vmap_excluded()