Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/cache_embedding/
Dcache_embedding.cc610 const CNodePtr &unique_node, const ParamSet &param_cache_enable_set) { in CacheEmbeddingForTrain()
676 const ParamSet &param_cache_enable_set) { in CacheEmbeddingForEval()
700 auto param_cache_enable_set = FindParamCacheEnable(graph); in AddCacheEmbedding() local