Home
last modified time | relevance | path

Searched refs:LocalDispatchKeySet (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/pytorch/c10/core/impl/
H A DLocalDispatchKeySet.h56 struct C10_API LocalDispatchKeySet { struct
57 /* implicit */ LocalDispatchKeySet(PODLocalDispatchKeySet x) in LocalDispatchKeySet() function
66 C10_API LocalDispatchKeySet tls_local_dispatch_key_set(); argument
70 inline C10_API LocalDispatchKeySet tls_local_dispatch_key_set() { in tls_local_dispatch_key_set()
78 C10_API void _force_tls_local_dispatch_key_set(LocalDispatchKeySet key_set);
122 ForceDispatchKeyGuard(c10::impl::LocalDispatchKeySet key_set) in ForceDispatchKeyGuard()
140 c10::impl::LocalDispatchKeySet saved_keyset_;
H A DLocalDispatchKeySet.cpp17 LocalDispatchKeySet tls_local_dispatch_key_set() { in tls_local_dispatch_key_set()
22 void _force_tls_local_dispatch_key_set(LocalDispatchKeySet key_set) { in _force_tls_local_dispatch_key_set()
/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A DPythonFallbackKernel.cpp17 thread_local std::optional<c10::impl::LocalDispatchKeySet> tls_on_entry;
19 c10::impl::LocalDispatchKeySet safe_get_tls_on_entry() { in safe_get_tls_on_entry()
45 c10::impl::LocalDispatchKeySet saved_;
H A DPythonFallbackKernel.h12 c10::impl::LocalDispatchKeySet saved_;
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DInterpreter.h146 void saveLocalDispatchKeySet(c10::impl::LocalDispatchKeySet keyset) { in saveLocalDispatchKeySet()
154 c10::impl::LocalDispatchKeySet getSavedLocalDispatchKeySet() const { in getSavedLocalDispatchKeySet()
182 std::optional<c10::impl::LocalDispatchKeySet> savedLocalDispatchKeySet_;
H A DDynamicLayer.cpp469 restoreLocalDispatchKeySetRAII(const c10::impl::LocalDispatchKeySet& key_set) { in restoreLocalDispatchKeySetRAII()
/aosp_15_r20/external/pytorch/c10/core/
H A DInferenceMode.h84 c10::impl::LocalDispatchKeySet prev_keyset;
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DThreadLocalState.h43 c10::impl::LocalDispatchKeySet dispatch_key_;
/aosp_15_r20/external/pytorch/torch/csrc/dynamo/
H A Dguards.h17 c10::impl::LocalDispatchKeySet dispatch_modifier;
/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dthroughput_benchmark-inl.h65 c10::impl::LocalDispatchKeySet tls_key_set = in benchmark()
/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()