Searched refs:getDispatchTableIndexForDispatchKey (Results 1 – 5 of 5) sorted by relevance
240 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()
424 auto idx = getDispatchTableIndexForDispatchKey(dispatchKey); in registerFallback()449 auto idx = getDispatchTableIndexForDispatchKey(dispatchKey); in deregisterFallback_()
192 auto dispatch_ix = getDispatchTableIndexForDispatchKey(dk); in hasBackendFallbackForDispatchKey()761 auto idx = getDispatchTableIndexForDispatchKey(dk);
282 auto idx = getDispatchTableIndexForDispatchKey(k); in _fallback()
633 C10_API inline int getDispatchTableIndexForDispatchKey(DispatchKey k) { in getDispatchTableIndexForDispatchKey() function