Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/c10/core/impl/
H A DLocalDispatchKeySet.cpp14 thread_local PODLocalDispatchKeySet raw_local_dispatch_key_set; variable
18 return raw_local_dispatch_key_set; in tls_local_dispatch_key_set()
23 raw_local_dispatch_key_set.set_included(key_set.included_); in _force_tls_local_dispatch_key_set()
24 raw_local_dispatch_key_set.set_excluded(key_set.excluded_); in _force_tls_local_dispatch_key_set()
49 : tls_(&raw_local_dispatch_key_set), include_(include - tls_->included()) { in IncludeDispatchKeyGuard()
62 : tls_(&raw_local_dispatch_key_set), exclude_(exclude - tls_->excluded()) { in ExcludeDispatchKeyGuard()
79 return raw_local_dispatch_key_set.excluded().has(x); in tls_is_dispatch_key_excluded()
83 auto* tls = &raw_local_dispatch_key_set; in tls_set_dispatch_key_excluded()
95 return raw_local_dispatch_key_set.included().has(x); in tls_is_dispatch_key_included()
99 auto* tls = &raw_local_dispatch_key_set; in tls_set_dispatch_key_included()
[all …]
H A DLocalDispatchKeySet.h68 extern C10_API thread_local PODLocalDispatchKeySet raw_local_dispatch_key_set;
73 return raw_local_dispatch_key_set; in tls_local_dispatch_key_set()