Searched defs:embedding_table (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/ |
| D | pserver_kernel.h | 41 …virtual void UpdateEmbeddings(float *embedding_table, const size_t *lookup_ids, const float *updat… in UpdateEmbeddings()
|
| D | embedding_look_up_ps_kernel.cc | 90 void EmbeddingLookUpPSKernel::UpdateEmbeddings(float *embedding_table, const size_t *lookup_ids, in UpdateEmbeddings()
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | parameter_server.cc | 395 AddressPtr embedding_table = std::make_shared<kernel::Address>(); in DoEmbeddingLookup() local
499 auto embedding_table = AnfAlgo::GetInputNode(cnode, 0); in GetEmbeddingTableParamPtr() local
528 auto embedding_table = AnfAlgo::GetInputNode(cnode, 0); in CacheEmbeddingTableParamPtr() local
545 for (auto embedding_table : embedding_tables_) { in SyncEmbeddingTables() local
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | embedding.py | 98 def __init__(self, vocab_size, embedding_size, use_one_hot=False, embedding_table='normal', argument
|
| D | thor_layer.py | 570 def __init__(self, vocab_size, embedding_size, use_one_hot=False, embedding_table='normal', argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | session_basic.cc | 2656 auto embedding_table = AnfAlgo::GetInputNode(node->cast<CNodePtr>(), embedding_table_idx); in AssignParamKey() local
|