Searched refs:tls_local_dispatch_key_set (Results 1 – 18 of 18) sorted by relevance
66 C10_API LocalDispatchKeySet tls_local_dispatch_key_set();70 inline C10_API LocalDispatchKeySet tls_local_dispatch_key_set() { in tls_local_dispatch_key_set() function121 : saved_keyset_(c10::impl::tls_local_dispatch_key_set()) {} in ForceDispatchKeyGuard()
17 LocalDispatchKeySet tls_local_dispatch_key_set() { in tls_local_dispatch_key_set() function
183 !c10::impl::tls_local_dispatch_key_set().excluded_.has( in _to_copy()290 if (c10::impl::tls_local_dispatch_key_set().excluded_.has( in empty_symint()480 auto curr_tls = c10::impl::tls_local_dispatch_key_set(); in logsumexp_out()
192 auto tmp = c10::impl::tls_local_dispatch_key_set(); in SaveLocalDispatchKeySet()401 auto tls = c10::impl::tls_local_dispatch_key_set(); in dump_local_tls()
59 auto local_keyset = c10::impl::tls_local_dispatch_key_set(); in setup_dispatch_key_tls()
56 prev_keyset(c10::impl::tls_local_dispatch_key_set()) { in prev_mode()
15 : dispatch_key_(c10::impl::tls_local_dispatch_key_set()), in ThreadLocalState()
262 if (!c10::impl::tls_local_dispatch_key_set().included_.has(c10::DispatchKey::Functionalize)) { in _to_copy_functionalize()
862 auto curr_tls = c10::impl::tls_local_dispatch_key_set(); in functionalize_op_helper()
30 : dispatch_modifier(c10::impl::tls_local_dispatch_key_set()), in LocalState()
809 return c10::impl::tls_local_dispatch_key_set().included_; in initDispatchBindings()812 return c10::impl::tls_local_dispatch_key_set().excluded_; in initDispatchBindings()934 auto include_set = c10::impl::tls_local_dispatch_key_set().included_; in initDispatchBindings()
66 c10::impl::tls_local_dispatch_key_set(); in benchmark()
140 tls_on_entry = c10::impl::tls_local_dispatch_key_set(); in MaybeSetTLSOnEntryGuard()
51 …return c10::impl::tls_local_dispatch_key_set().excluded_.isSupersetOf(c10::autograd_dispatch_keyse… in variable_excluded_from_dispatch()
40 c10::impl::LocalDispatchKeySet local = c10::impl::tls_local_dispatch_key_set(); in computeDispatchKeySet()
2242 auto local_keyset = c10::impl::tls_local_dispatch_key_set(); in initModule()2254 auto local_keyset = c10::impl::tls_local_dispatch_key_set(); in initModule()2259 auto local_keyset = c10::impl::tls_local_dispatch_key_set(); in initModule()
91 c10::impl::tls_local_dispatch_key_set().excluded_.has( in create_nested_view_tensor()
399 auto tls = c10::impl::tls_local_dispatch_key_set(); in dump_local_tls()