Searched refs:_force_tls_local_dispatch_key_set (Results 1 – 8 of 8) sorted by relevance
78 C10_API void _force_tls_local_dispatch_key_set(LocalDispatchKeySet key_set);124 c10::impl::_force_tls_local_dispatch_key_set(key_set); in ForceDispatchKeyGuard()133 c10::impl::_force_tls_local_dispatch_key_set(updated_set); in ForceDispatchKeyGuard()136 c10::impl::_force_tls_local_dispatch_key_set(saved_keyset_); in ~ForceDispatchKeyGuard()
22 void _force_tls_local_dispatch_key_set(LocalDispatchKeySet key_set) { in _force_tls_local_dispatch_key_set() function
73 c10::impl::_force_tls_local_dispatch_key_set(cur_keyset); in prev_mode()78 c10::impl::_force_tls_local_dispatch_key_set(prev_keyset); in ~InferenceMode()
57 c10::impl::_force_tls_local_dispatch_key_set(state.dispatch_key_); in setThreadLocalState()
65 c10::impl::_force_tls_local_dispatch_key_set(local_keyset); in setup_dispatch_key_tls()
201 c10::impl::_force_tls_local_dispatch_key_set(tmp); in ~SaveLocalDispatchKeySet()
73 c10::impl::_force_tls_local_dispatch_key_set(tls_key_set); in benchmark()
2250 c10::impl::_force_tls_local_dispatch_key_set(local_keyset); in initModule()