Searched defs:DispatchKeySet (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/c10/test/core/ |
| H A D | DispatchKeySet_test.cpp | 14 TEST(DispatchKeySet, ShowSemantics) { in TEST() argument
119 TEST(DispatchKeySet, Empty) { in TEST() argument
137 TEST(DispatchKeySet, SingletonBackendComponent) { in TEST() argument
158 TEST(DispatchKeySet, SingletonFunctionalityKeys) { in TEST() argument
176 TEST(DispatchKeySet, SingletonPerBackendFunctionalityKeys) { in TEST() argument
212 TEST(DispatchKeySet, DoubletonPerBackend) { in TEST() argument
293 TEST(DispatchKeySet, Full) { in TEST() argument
302 TEST(DispatchKeySet, IteratorBasicOps) { in TEST() argument
320 TEST(DispatchKeySet, getHighestPriorityBackendTypeId) { in TEST() argument
345 TEST(DispatchKeySet, IteratorEmpty) { in TEST() argument
[all …]
|
| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | DispatchKeySet.h | 175 constexpr DispatchKeySet(Full) in DispatchKeySet() function
178 constexpr DispatchKeySet(FullAfter, DispatchKey t) in DispatchKeySet() function
194 constexpr DispatchKeySet(Raw, uint64_t x) : repr_(x) {} in DispatchKeySet() function
196 constexpr explicit DispatchKeySet(BackendComponent k) { in DispatchKeySet() function
204 constexpr explicit DispatchKeySet(DispatchKey k) { in DispatchKeySet() function
267 explicit constexpr DispatchKeySet(std::initializer_list<DispatchKey> ks) in DispatchKeySet() function
270 explicit constexpr DispatchKeySet(std::initializer_list<BackendComponent> ks) in DispatchKeySet() function
509 constexpr DispatchKeySet(uint64_t repr) : repr_(repr) {} in DispatchKeySet() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/boxing/ |
| H A D | KernelFunction.cpp | 13 void fallthrough_kernel(OperatorKernel*, const OperatorHandle&, DispatchKeySet, Stack*) { in fallthrough_kernel() argument
22 void ambiguous_autogradother_kernel(OperatorKernel*, const OperatorHandle& op, DispatchKeySet, Stac… in ambiguous_autogradother_kernel() argument
35 void named_not_supported_kernel(OperatorKernel*, const OperatorHandle& op, DispatchKeySet, Stack*) { in named_not_supported_kernel() argument
|
| H A D | BoxedKernel_impl.h | 16 …make_boxed_function(OperatorKernel*, const OperatorHandle& opHandle, DispatchKeySet, Stack* stack)… in make_boxed_function() argument
|