Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/map_tensor/
Dmap_tensor_put_cpu_kernel.cc101 auto embedding_storage = embedding_storage_manager.Get(parameter_key_); in LaunchKernel() local
Dmap_tensor_get_cpu_kernel.cc103 auto embedding_storage = embedding_storage_manager.Get(parameter_key_); in LaunchKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dembedding_look_up_cpu_kernel.cc177 auto embedding_storage = embedding_storage_manager.Get(parameter_key_); in LaunchKernel() local
Dscatter_arithmetic_cpu_kernel.cc140 auto embedding_storage = embedding_storage_manager.Get(parameter_key_); in LaunchKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/distributed/embedding_cache/
Dembedding_cache_utils.cc468 std::shared_ptr<storage::AbstractEmbeddingStorage> embedding_storage = nullptr; in CreateEmbeddingStorageFunc() local
557 const auto &embedding_storage = item.second; in Clear() local