Searched defs:device_mem_addr (Results 1 – 4 of 4) sorted by relevance
188 Status EmbeddingCache::SetDeviceCacheAddr(void *device_mem_addr, size_t size) { in SetDeviceCacheAddr()
150 …gCacheManager::SetDeviceCacheAddr(const std::string &tensor_name, void *device_mem_addr, size_t si… in SetDeviceCacheAddr()
187 Status EmbeddingCache::SetDeviceCacheAddr(void *device_mem_addr, size_t size) { in SetDeviceCacheAddr()