Home
last modified time | relevance | path

Searched refs:tls_reenable_functionalize (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/
H A Dts_native_functions.cpp481 auto tls_reenable_functionalize = c10::impl::PODLocalDispatchKeySet(); in logsumexp_out() local
482 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()
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DFunctionalTensorWrapper.cpp863 auto tls_reenable_functionalize = c10::impl::PODLocalDispatchKeySet(); in functionalize_op_helper() local
864 tls_reenable_functionalize.set_included(curr_tls.included_); in functionalize_op_helper()
865tls_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()