Home
last modified time | relevance | path

Searched defs:embedding_host_cache (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/embedding_cache/
Ddevice_sparse_embedding_operation.cc37 auto embedding_host_cache = cache_analysis->embedding_host_cache_; in PushCacheFromDeviceToLocalHost() local
90 auto embedding_host_cache = cache_analysis->embedding_host_cache_; in PullCacheFromLocalHostToDevice() local
Ddevice_embedding_operation.cc35 EmbeddingHostCache *embedding_host_cache, in ParseHostDataHostToDevice()
112 EmbeddingHostCache *embedding_host_cache, in ParseHostDataDeviceToHost()
Ddevice_dense_embedding_operation.cc32 EmbeddingHostCache *embedding_host_cache, in AnalyseCache()
89 auto embedding_host_cache = cache_analysis->embedding_host_cache_; in PushCacheFromDeviceToLocalHost() local
142 auto embedding_host_cache = cache_analysis->embedding_host_cache_; in PullCacheFromLocalHostToDevice() local
Ddevice_embedding_operation.h77 … EmbeddingDeviceCache *embedding_device_cache, EmbeddingHostCache *embedding_host_cache, in AnalyseCache()
Dembedding_cache_prefetch_actor.cc534 … EmbeddingHostCache *embedding_host_cache = new (std::nothrow) EmbeddingHostCache(unique_ids_num); in AnalyseCacheTask() local
753 auto embedding_host_cache = cache_analysis->embedding_host_cache_; in PushCacheFromLocalHostToRemote() local
785 auto embedding_host_cache = cache_analysis->embedding_host_cache_; in PullCacheFromRemoteToLocalHost() local
817 auto embedding_host_cache = cache_analysis->embedding_host_cache_; in InitLocalCacheForNewIds() local