Home
last modified time | relevance | path

Searched refs:AddWeightParameter (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/cache_embedding/
Dcache_embedding.cc55 auto cache_param = graph->AddWeightParameter(cache_name); in AddCacheParameters()
259 auto hashmap = func_graph->AddWeightParameter(hashmap_name); in InitHashMap()
272 auto step = func_graph->AddWeightParameter(step_name); in InitStep()
539 auto new_param = graph->AddWeightParameter(new_param_name); in CreateOutputNodeParam()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/parse/
Dresolve.cc124 auto node = top_func_graph->AddWeightParameter(param_name); in ResolveParameterObj()
/third_party/mindspore/mindspore/core/ir/
Dfunc_graph.h184 ParameterPtr AddWeightParameter(const std::string &name);
Dfunc_graph.cc116 ParameterPtr FuncGraph::AddWeightParameter(const std::string &name) { in AddWeightParameter() function in mindspore::FuncGraph