Home
last modified time | relevance | path

Searched defs:sparse_gatherv2_with_cache (Results 1 – 1 of 1) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/cache_embedding/
Dcache_embedding.cc624 CNodePtrList sparse_gatherv2_with_cache = FindSparseGatherV2WithCache(cnodes, param_set); in CacheEmbeddingForTrain() local
684 auto sparse_gatherv2_with_cache = FindSparseGatherV2WithCache(cnodes, param_cache_enable_set); in CacheEmbeddingForEval() local