Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/tests/ut/cpp/distributed/embedding_cache/
Dsparse_embedding_storage_test.cc45 int32_t embedding_key = 0; in TEST_F() local
114 int32_t embedding_key = 0; in TEST_F() local
Ddense_embedding_storage_test.cc42 int32_t embedding_key = 0; in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/distributed/embedding_cache/embedding_storage/
Dembedding_storage.h49 …: embedding_key_(embedding_key), embedding_dim_(embedding_dim), cache_capacity_(cache_capacity), a… in embedding_key_() argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/distributed/embedding_cache/
Dembedding_cache_utils.cc467 void CreateEmbeddingStorageFunc(int32_t embedding_key, size_t embedding_dim, size_t capacity) { in CreateEmbeddingStorageFunc()
513 void CreateEmbeddingStorage(std::pair<TypeId, TypeId> key_value_types, int32_t embedding_key, size_… in CreateEmbeddingStorage()