| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/distributed/embedding_cache/ |
| D | sparse_embedding_storage_test.cc | 46 size_t embedding_dim = 8; in TEST_F() local
115 size_t embedding_dim = 1024; in TEST_F() local
|
| D | dense_embedding_storage_test.cc | 43 size_t embedding_dim = 8; in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/manually_defined/ |
| D | ops_def.py | 1200 def __init__(self, embedding_dim, value_total_len, export_mode="all", argument
1213 def __init__(self, embedding_dim, value_total_len, argument
1246 def __init__(self, value_total_len, embedding_dim, _table_id, argument
1254 def init_embedding_hashmap(table_id, value_total_len, embedding_dim, _table_id, argument
1293 def __init__(self, embedding_dim, _max_key_num, mask_zero=(0,), argument
1304 def __init__(self, embedding_dim, _max_key_num, amsgrad=(0,), argument
1315 def __init__(self, embedding_dim, _max_key_num, mask_zero=(0,), argument
1326 def __init__(self, embedding_dim, _max_key_num, mask_zero=(0,), argument
1337 def __init__(self, embedding_dim, _embedding_dim, _max_key_num, argument
1343 def embedding_table_find(table_id, keys, embedding_dim, _max_key_num, argument
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/extend/ |
| D | embedding.py | 92 … def __init__(self, num_embeddings, embedding_dim, padding_idx=None, max_norm=None, norm_type=2.0, argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/es/ |
| D | test_es_external_api.py | 30 def __init__(self, embedding_dim, max_feature_count, table_id_dict=None, es_initializer=None, argument
|
| D | test_ops_es.py | 27 embedding_dim = 12 variable
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/distributed/persistent/ |
| D | local_file_storage_test.cc | 56 size_t embedding_dim = 8; in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/ |
| D | embedding_table_find.cc | 68 for (const auto &embedding_dim : embedding_dims) { in InferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/layer/ |
| D | embedding_service_layer.py | 72 …(self, ps_num, ps_ids, train_mode, train_level, table_id, bucket_size, embedding_dim, slot_var_num, argument
151 def __init__(self, table_id, es_initializer, embedding_dim, max_key_num, optimizer_mode=None, argument
|
| D | embedding_service.py | 100 def check_common_init_params(name, init_vocabulary_size, embedding_dim): argument
199 def embedding_init(self, name, init_vocabulary_size, embedding_dim, max_feature_count, argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/embedding_cache/ |
| D | embedding_cache_prefetch_actor.cc | 954 size_t embedding_dim = outputs->size() / ids_num; in PullEembeddingsFromRemote() local
1009 size_t embedding_dim = (embeddings_len / ids_num) / sizeof(float); in DoPushEmbeddingsToRemote() local
1131 size_t embedding_dim = (embeddings_len / ids_num) / sizeof(float); in PartitionIdsAndEmbeddings() local
1152 … size_t server_rank_id, size_t embedding_dim, const void *keys, in SendToRemote()
1228 size_t embedding_dim = outputs->size() / ids_num; in RetrieveEmbeddings() local
1384 size_t embedding_dim = 1; in FinalizeRemote() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/distributed/embedding_cache/ |
| D | embedding_cache_utils.cc | 467 void CreateEmbeddingStorageFunc(int32_t embedding_key, size_t embedding_dim, size_t capacity) { in CreateEmbeddingStorageFunc()
513 …dingStorage(std::pair<TypeId, TypeId> key_value_types, int32_t embedding_key, size_t embedding_dim, in CreateEmbeddingStorage()
|