Searched refs:kEmbeddingLookupOpName (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/ps/ps_cache/ascend/ |
| D | ascend_ps_cache.cc | 219 hash_swap_out_mod->SetNodeName(kEmbeddingLookupOpName); in HashSwapOut()
232 …std::make_shared<KernelNodeInfo>(kEmbeddingLookupOpName, input_shape, input_type, output_shape, ou… in HashSwapOut()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/ |
| D | replace_node_by_proxy.cc | 61 … (node != nullptr && node->isa<CNode>() && AnfAlgo::GetCNodeName(node) == kEmbeddingLookupOpName) { in Run()
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | utils.h | 239 constexpr auto kEmbeddingLookupOpName = "EmbeddingLookup"; variable
685 kCastOpName, kExpandDimsOpName, kReshapeOpName, kEmbeddingLookupOpName, kReduceMinOpName,
689 …kCastOpName, kExpandDimsOpName, kReshapeOpName, kEmbeddingLookupOpName, kTransposeOpName, k…
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/cache_embedding/ |
| D | cache_embedding.cc | 365 PrimitivePtr emb_lookup_primitive = std::make_shared<Primitive>(kEmbeddingLookupOpName); in CreateEmbeddingLookup()
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | parameter_server.cc | 498 …if (cnode_name == kEmbeddingLookupOpName || cnode_name == kGatherV2OpName || cnode_name == kSparse… in GetEmbeddingTableParamPtr()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | session_basic.cc | 2654 if (AnfAlgo::GetCNodeName(node) == kEmbeddingLookupOpName) { in AssignParamKey()
|