Searched refs:old_emb_idx (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/cache_embedding/ |
| D | cache_embedding.cc | 327 auto old_emb_idx = std::make_shared<abstract::AbstractTensor>( in CreateMapCacheIdx() local
334 …std::vector<std::shared_ptr<abstract::AbstractBase>> elements = {cache_idx, old_emb_idx, miss_emb_… in CreateMapCacheIdx()
383 AnfNodePtr CreateUpdateCache(const FuncGraphPtr &graph, ParameterPtr params, AnfNodePtr old_emb_idx, in CreateUpdateCache() argument
400 …nfNodePtr> update_cache_nodes{NewValueNode(update_cache_primitive), params, old_emb_idx, old_value, in CreateUpdateCache()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | prim_arrays.cc | 455 auto old_emb_idx = in InferImplMapCacheIdx() local
462 AbstractBasePtrList elements = {cache_idx, old_emb_idx, miss_emb_idx, swap_emb_idx}; in InferImplMapCacheIdx()
|