Searched refs:ForceDispatchKeyGuard (Results 1 – 6 of 6) sorted by relevance
118 struct C10_API ForceDispatchKeyGuard { struct120 ForceDispatchKeyGuard() in ForceDispatchKeyGuard() function122 ForceDispatchKeyGuard(c10::impl::LocalDispatchKeySet key_set) in ForceDispatchKeyGuard() function123 : ForceDispatchKeyGuard() { in ForceDispatchKeyGuard()126 ForceDispatchKeyGuard( in ForceDispatchKeyGuard() function129 : ForceDispatchKeyGuard() { in ForceDispatchKeyGuard()135 ~ForceDispatchKeyGuard() { in ~ForceDispatchKeyGuard() argument
13 c10::impl::ForceDispatchKeyGuard guard_;
468 static c10::impl::ForceDispatchKeyGuard470 return c10::impl::ForceDispatchKeyGuard(key_set); in restoreLocalDispatchKeySetRAII()
830 c10::impl::ForceDispatchKeyGuard, in initDispatchBindings()833 py_context_manager<c10::impl::ForceDispatchKeyGuard>( in initDispatchBindings()
485 c10::impl::ForceDispatchKeyGuard guard_(tls_reenable_functionalize); in logsumexp_out()
866 c10::impl::ForceDispatchKeyGuard guard_(tls_reenable_functionalize); in functionalize_op_helper()