Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/core/utils/
Dflags.h26 inline const char GRAPH_FLAG_CACHE_ENABLE[] = "cache_enable"; variable
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/cache_embedding/
Dcache_embedding.cc627 graph->set_flag(GRAPH_FLAG_CACHE_ENABLE, true); in CacheEmbeddingForTrain()
686 graph->set_flag(GRAPH_FLAG_CACHE_ENABLE, true); in CacheEmbeddingForEval()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dpass.cc597 if (func_graph->has_flag(GRAPH_FLAG_CACHE_ENABLE)) { in AddCacheEmbeddingPass()