Searched refs:num_functionality_keys (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | DispatchKeySet.cpp | 259 std::array<FunctionalityOffsetAndMask, num_functionality_keys>
261 std::array<FunctionalityOffsetAndMask, num_functionality_keys> in initializeFunctionalityOffsetsAndMasks()
266 for (const auto functionality_idx : c10::irange(1, num_functionality_keys)) { in initializeFunctionalityOffsetsAndMasks()
288 offsets_and_masks[num_functionality_keys - 1].offset == in initializeFunctionalityOffsetsAndMasks()
293 offsets_and_masks[num_functionality_keys - 1].offset); in initializeFunctionalityOffsetsAndMasks()
|
| H A D | DispatchKeySet.h | 38 C10_API std::array<FunctionalityOffsetAndMask, num_functionality_keys>
42 array<FunctionalityOffsetAndMask, num_functionality_keys>&
176 : repr_((1ULL << (num_backends + num_functionality_keys - 1)) - 1) {} in DispatchKeySet()
534 num_backends + num_functionality_keys;
536 static const uint8_t end_iter_key_val = num_functionality_keys;
|
| H A D | DispatchKey.h | 562 constexpr uint8_t num_functionality_keys = variable
578 for (uint8_t k = 0; k <= num_functionality_keys; ++k) { in numPerBackendFunctionalityKeys()
589 constexpr uint16_t num_runtime_entries = num_functionality_keys +
|
| /aosp_15_r20/external/pytorch/c10/test/core/ |
| H A D | DispatchKeySet_test.cpp | 159 for (const auto i : c10::irange(1, num_functionality_keys)) { in TEST()
295 for (const auto i : c10::irange(1, num_functionality_keys)) { in TEST()
389 DispatchKeySet::iterator(&raw_repr, num_backends + num_functionality_keys); in TEST()
393 &raw_repr, num_backends + num_functionality_keys + 1), in TEST()
|