Home
last modified time | relevance | path

Searched refs:tls_local_dispatch_key_set (Results 1 – 18 of 18) sorted by relevance

/aosp_15_r20/external/pytorch/c10/core/impl/
H A DLocalDispatchKeySet.h66 C10_API LocalDispatchKeySet tls_local_dispatch_key_set();
70 inline C10_API LocalDispatchKeySet tls_local_dispatch_key_set() { in tls_local_dispatch_key_set() function
121 : saved_keyset_(c10::impl::tls_local_dispatch_key_set()) {} in ForceDispatchKeyGuard()
H A DLocalDispatchKeySet.cpp17 LocalDispatchKeySet tls_local_dispatch_key_set() { in tls_local_dispatch_key_set() function
/aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/
H A Dts_native_functions.cpp183 !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()
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DDynamicLayer.cpp192 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()
H A DInterpreter.cpp59 auto local_keyset = c10::impl::tls_local_dispatch_key_set(); in setup_dispatch_key_tls()
/aosp_15_r20/external/pytorch/c10/core/
H A DInferenceMode.h56 prev_keyset(c10::impl::tls_local_dispatch_key_set()) { in prev_mode()
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DThreadLocalState.cpp15 : dispatch_key_(c10::impl::tls_local_dispatch_key_set()), in ThreadLocalState()
H A DFunctionalizeFallbackKernel.cpp262 if (!c10::impl::tls_local_dispatch_key_set().included_.has(c10::DispatchKey::Functionalize)) { in _to_copy_functionalize()
H A DFunctionalTensorWrapper.cpp862 auto curr_tls = c10::impl::tls_local_dispatch_key_set(); in functionalize_op_helper()
/aosp_15_r20/external/pytorch/torch/csrc/dynamo/
H A Dguards.h30 : dispatch_modifier(c10::impl::tls_local_dispatch_key_set()), in LocalState()
/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dpython_dispatch.cpp809 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()
H A Dthroughput_benchmark-inl.h66 c10::impl::tls_local_dispatch_key_set(); in benchmark()
/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A DPythonFallbackKernel.cpp140 tls_on_entry = c10::impl::tls_local_dispatch_key_set(); in MaybeSetTLSOnEntryGuard()
H A DTensorBase.h51 …return c10::impl::tls_local_dispatch_key_set().excluded_.isSupersetOf(c10::autograd_dispatch_keyse… in variable_excluded_from_dispatch()
/aosp_15_r20/external/pytorch/aten/src/ATen/core/dispatch/
H A DDispatchKeyExtractor.h40 c10::impl::LocalDispatchKeySet local = c10::impl::tls_local_dispatch_key_set(); in computeDispatchKeySet()
/aosp_15_r20/external/pytorch/torch/csrc/
H A DModule.cpp2242 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()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/
H A DNestedTensorUtils.h91 c10::impl::tls_local_dispatch_key_set().excluded_.has( in create_nested_view_tensor()
/aosp_15_r20/external/pytorch/torch/csrc/functorch/
H A Dinit.cpp399 auto tls = c10::impl::tls_local_dispatch_key_set(); in dump_local_tls()