Searched refs:func_graph_user (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/core/ir/ |
| D | func_graph_cloner.cc | 378 void Cloner::AddInputs(const FuncGraphPtr &func_graph_user, const FuncGraphPtr &func_graph, in AddInputs() argument
381 auto &repl_func_graph = repl_map_func_graph_[func_graph_user]; in AddInputs()
384 node = func_graph_user->NewCNode({NewValueNode(prim::kPrimPartial), NewValueNode(func_graph)}); in AddInputs()
482 void Cloner::LiftParameters(const FuncGraphPtr &func_graph_user, const FuncGraphPtr &func_graph, in LiftParameters() argument
486 AddParameters(func_graph_user, params, &lift_params, &input_params); in LiftParameters()
487 AddInputs(func_graph_user, func_graph, input_params); in LiftParameters()
491 for (auto &cnode : func_graph_user->func_graph_cnodes_index()) { in LiftParameters()
492 LiftParameters(cnode.first->first->func_graph(), func_graph_user, lift_params); in LiftParameters()
|
| D | func_graph_cloner.h | 97 …void AddInputs(const FuncGraphPtr &func_graph_user, const FuncGraphPtr &func_graph, const AnfNodeP…
100 void LiftParameters(const FuncGraphPtr &func_graph_user, const FuncGraphPtr &func_graph,
|