Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ops/_op_impl/aicpu/
D__init__.py62 from .cache_swap_table import _cache_swap_table_aicpu
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/cache_embedding/
Dcache_embedding.cc379 auto cache_swap_table = graph->NewCNode(cache_swap_table_nodes); in CreateCacheSwapTable() local
380 return cache_swap_table; in CreateCacheSwapTable()
412 …auto cache_swap_table = CreateCacheSwapTable(graph, ele.first, map_cache_idx_node_outputs[3], emb_… in CreateEmbSwapUpdate() local
413 …date_cache = CreateUpdateCache(graph, ele.second, map_cache_idx_node_outputs[1], cache_swap_table); in CreateEmbSwapUpdate()
414 node_pair_list.emplace_back(std::make_pair(cache_swap_table, update_cache)); in CreateEmbSwapUpdate()