Searched refs:tls_reenable_functionalize (Results 1 – 2 of 2) sorted by relevance
481 auto tls_reenable_functionalize = c10::impl::PODLocalDispatchKeySet(); in logsumexp_out() local482 tls_reenable_functionalize.set_included(curr_tls.included_); in logsumexp_out()483 tls_reenable_functionalize.set_excluded( in logsumexp_out()485 c10::impl::ForceDispatchKeyGuard guard_(tls_reenable_functionalize); in logsumexp_out()
863 auto tls_reenable_functionalize = c10::impl::PODLocalDispatchKeySet(); in functionalize_op_helper() local864 tls_reenable_functionalize.set_included(curr_tls.included_); in functionalize_op_helper()865 …tls_reenable_functionalize.set_excluded(curr_tls.excluded_.remove(c10::DispatchKey::Functionalize)… in functionalize_op_helper()866 c10::impl::ForceDispatchKeyGuard guard_(tls_reenable_functionalize); in functionalize_op_helper()