Searched refs:LocalDispatchKeySet (Results 1 – 11 of 11) sorted by relevance
56 struct C10_API LocalDispatchKeySet { struct57 /* implicit */ LocalDispatchKeySet(PODLocalDispatchKeySet x) in LocalDispatchKeySet() function66 C10_API LocalDispatchKeySet tls_local_dispatch_key_set(); argument70 inline C10_API LocalDispatchKeySet tls_local_dispatch_key_set() { in tls_local_dispatch_key_set()78 C10_API void _force_tls_local_dispatch_key_set(LocalDispatchKeySet key_set);122 ForceDispatchKeyGuard(c10::impl::LocalDispatchKeySet key_set) in ForceDispatchKeyGuard()140 c10::impl::LocalDispatchKeySet saved_keyset_;
17 LocalDispatchKeySet tls_local_dispatch_key_set() { in tls_local_dispatch_key_set()22 void _force_tls_local_dispatch_key_set(LocalDispatchKeySet key_set) { in _force_tls_local_dispatch_key_set()
17 thread_local std::optional<c10::impl::LocalDispatchKeySet> tls_on_entry;19 c10::impl::LocalDispatchKeySet safe_get_tls_on_entry() { in safe_get_tls_on_entry()45 c10::impl::LocalDispatchKeySet saved_;
12 c10::impl::LocalDispatchKeySet saved_;
146 void saveLocalDispatchKeySet(c10::impl::LocalDispatchKeySet keyset) { in saveLocalDispatchKeySet()154 c10::impl::LocalDispatchKeySet getSavedLocalDispatchKeySet() const { in getSavedLocalDispatchKeySet()182 std::optional<c10::impl::LocalDispatchKeySet> savedLocalDispatchKeySet_;
469 restoreLocalDispatchKeySetRAII(const c10::impl::LocalDispatchKeySet& key_set) { in restoreLocalDispatchKeySetRAII()
84 c10::impl::LocalDispatchKeySet prev_keyset;
43 c10::impl::LocalDispatchKeySet dispatch_key_;
17 c10::impl::LocalDispatchKeySet dispatch_modifier;
65 c10::impl::LocalDispatchKeySet tls_key_set = in benchmark()
40 c10::impl::LocalDispatchKeySet local = c10::impl::tls_local_dispatch_key_set(); in computeDispatchKeySet()