Home
last modified time | relevance | path

Searched defs:host_cache_tensor (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/parameter_cache/
Dembedding_cache.cc64 Status EmbeddingCache::Init(mindspore::MSTensor host_cache_tensor, mindspore::MSTensor device_tenso… in Init()
125 …mbeddingCache::Init(uint32_t device_id, const void *context, mindspore::MSTensor host_cache_tensor, in Init()
Dembedding_cache_manager.cc88 auto host_cache_tensor = host_cache_model_->GetHostCacheTensor(kernel); in InitCacheKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/parameter_cache/
Dembedding_cache.cc63 Status EmbeddingCache::Init(mindspore::MSTensor host_cache_tensor, mindspore::MSTensor device_tenso… in Init()
124 …mbeddingCache::Init(uint32_t device_id, const void *context, mindspore::MSTensor host_cache_tensor, in Init()
Dembedding_cache_manager.cc88 auto host_cache_tensor = host_cache_model_->GetHostCacheTensor(kernel); in InitCacheKernel() local