Home
last modified time | relevance | path

Searched defs:PushCacheFromDeviceToLocalHost (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/embedding_cache/
Ddevice_sparse_embedding_operation.cc32 bool DeviceSparseEmbeddingOperation::PushCacheFromDeviceToLocalHost(const HashTableInfo &hash_info, in PushCacheFromDeviceToLocalHost() function in mindspore::runtime::DeviceSparseEmbeddingOperation
Ddevice_dense_embedding_operation.cc84 bool DeviceDenseEmbeddingOperation::PushCacheFromDeviceToLocalHost(const HashTableInfo &hash_info, in PushCacheFromDeviceToLocalHost() function in mindspore::runtime::DeviceDenseEmbeddingOperation