Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/core/dispatch/
H A DOperatorEntry.cpp240 const auto dispatch_ix = getDispatchTableIndexForDispatchKey(k); in registerKernel()
372 auto dispatch_ix = getDispatchTableIndexForDispatchKey(dispatch_key); in registerKernel()
391 const auto dispatch_ix = getDispatchTableIndexForDispatchKey(dispatch_key); in registerKernel()
472 auto idx = getDispatchTableIndexForDispatchKey(k); in registerKernel()
488 auto iter = getDispatchTableIndexForDispatchKey(k); in registerKernel()
H A DDispatcher.cpp424 auto idx = getDispatchTableIndexForDispatchKey(dispatchKey); in registerFallback()
449 auto idx = getDispatchTableIndexForDispatchKey(dispatchKey); in deregisterFallback_()
H A DDispatcher.h192 auto dispatch_ix = getDispatchTableIndexForDispatchKey(dk); in hasBackendFallbackForDispatchKey()
761 auto idx = getDispatchTableIndexForDispatchKey(dk);
/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A Dlibrary.cpp282 auto idx = getDispatchTableIndexForDispatchKey(k); in _fallback()
/aosp_15_r20/external/pytorch/c10/core/
H A DDispatchKeySet.h633 C10_API inline int getDispatchTableIndexForDispatchKey(DispatchKey k) { in getDispatchTableIndexForDispatchKey() function