Home
last modified time | relevance | path

Searched refs:emb_lookup (Results 1 – 1 of 1) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/cache_embedding/
Dcache_embedding.cc369 auto emb_lookup = graph->NewCNode(emb_lookup_nodes); in CreateEmbeddingLookup() local
370 return emb_lookup; in CreateEmbeddingLookup()
411 auto emb_lookup = CreateEmbeddingLookup(graph, ele.second, map_cache_idx_node_outputs[2]); in CreateEmbSwapUpdate() local
412 …che_swap_table = CreateCacheSwapTable(graph, ele.first, map_cache_idx_node_outputs[3], emb_lookup); in CreateEmbSwapUpdate()