Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/embedding_cache/
Ddevice_dense_embedding_operation.cc31 … int *indices, EmbeddingDeviceCache *embedding_device_cache, in AnalyseCache()
88 auto embedding_device_cache = cache_analysis->embedding_device_cache_; in PushCacheFromDeviceToLocalHost() local
141 auto embedding_device_cache = cache_analysis->embedding_device_cache_; in PullCacheFromLocalHostToDevice() local
475 EmbeddingDeviceCache *embedding_device_cache, in ParseDeviceData()
Ddevice_sparse_embedding_operation.cc36 auto embedding_device_cache = cache_analysis->embedding_device_cache_; in PushCacheFromDeviceToLocalHost() local
89 auto embedding_device_cache = cache_analysis->embedding_device_cache_; in PullCacheFromLocalHostToDevice() local
Ddevice_embedding_operation.h77 … EmbeddingDeviceCache *embedding_device_cache, EmbeddingHostCache *embedding_host_cache, in AnalyseCache()
Ddevice_embedding_operation.cc111 … EmbeddingDeviceCache *embedding_device_cache, in ParseHostDataDeviceToHost()
Dembedding_cache_prefetch_actor.cc532 …EmbeddingDeviceCache *embedding_device_cache = new (std::nothrow) EmbeddingDeviceCache(unique_ids_… in AnalyseCacheTask() local