Home
last modified time | relevance | path

Searched refs:isPerBackendFunctionalityKey (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/c10/core/
H A DDispatchKeySet.cpp215 if (isPerBackendFunctionalityKey( in operator ++()
280 auto next_mask = isPerBackendFunctionalityKey(k) ? full_backend_mask : 0; in initializeFunctionalityOffsetsAndMasks()
H A DDispatchKey.h547 constexpr bool isPerBackendFunctionalityKey(DispatchKey k) { in isPerBackendFunctionalityKey() function
579 if (isPerBackendFunctionalityKey(static_cast<DispatchKey>(k))) in numPerBackendFunctionalityKeys()
H A DDispatchKeySet.h430 if (isPerBackendFunctionalityKey(functionality_k)) { in highestPriorityTypeId()
584 if (isPerBackendFunctionalityKey(functionality_key)) {
/aosp_15_r20/external/pytorch/aten/src/ATen/core/dispatch/
H A DDispatchKeyExtractor.cpp16 if (isPerBackendFunctionalityKey(toFunctionalityKey(k))) { in setOperatorHasFallthroughForKey()
/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dpython_dispatch.cpp686 if (c10::isPerBackendFunctionalityKey(key)) { in initDispatchBindings()