Home
last modified time | relevance | path

Searched refs:set_func_graph (Results 1 – 16 of 16) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dpass.cc74 res->set_func_graph(new_fg); in DoRenormalize()
98 res->set_func_graph(func_graph); in TransformTopGraphPass()
559 res->set_func_graph(g_pass_opts[name]->step(func_graph)); in OptPassGroup()
647 res->set_func_graph(new_fg); in CconvPass()
684 res->set_func_graph(func_graph); in AutoMonadElimOptPass()
Dresource.h73 void set_func_graph(const FuncGraphPtr &func_graph) { func_graph_ = func_graph; } in set_func_graph() function
Daction.cc254 res->set_func_graph(ret); in Renormalize()
371 res->set_func_graph(top_graph); in ParseAction()
537 res->set_func_graph(new_fg); in AbstractSpecializeAction()
964 res->set_func_graph(fg); in SetMindIRGraphAction()
1048 res->set_func_graph(new_fg); in OptActionVmPyStub()
1069 res->set_func_graph(new_fg); in OptActionGePyStub()
Dpipeline.cc207 resource->set_func_graph(fg); in GetCachedFuncGraph()
/third_party/mindspore/mindspore/lite/tools/converter/parser/tf/
Dfunctionalize_cond.cc88 node->set_func_graph(graph); in BranchSubGraphAddNodes()
171 return_cnode->set_func_graph(graph); in CreateBranchGraph()
Dfunctionalize_while.cc319 node->set_func_graph(cond_sub_func_graph_); in CondSubgraphAddNodes()
455 node->set_func_graph(body_sub_func_graph_); in BodySubgraphAddNodes()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/
Dpy_pass.cc100 para_node->set_func_graph(top_graph); in BuildNewParameter()
272 para_node->set_func_graph(func_graph); in Run()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/
Dprim_bprop_optimizer.cc276 prim_bprop_opt_res->set_func_graph(graph_for_cache); in PrimBpropOptStep1()
302 resource->set_func_graph(bprop_fg); in PrimBpropOptStep2()
/third_party/mindspore/mindspore/lite/tools/common/
Dfunc_graph_subgraph.cc535 input->set_func_graph(sub_graph); in CreateParameterForPartialSubGraph()
551 node->set_func_graph(sub_graph); in CreateCNodeForPartialSubGraph()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dgraph_kernel_splitter.cc259 node->set_func_graph(func_graph); in CreateReturnNode()
534 node->set_func_graph(sub_func_graph); in BindFuncGraph()
573 cnode->set_func_graph(main_func_graph_); in InlineSubFuncGraph()
/third_party/mindspore/mindspore/core/ir/
Danf.h122 …virtual void set_func_graph(const FuncGraphPtr &func_graph) { func_graph_ = FuncGraphWeakPtr(func_… in set_func_graph() function
492 void set_func_graph(const FuncGraphPtr &func_graph) override { in set_func_graph() function
Dmanager.cc667 node->set_func_graph(target); in MoveAllCNodeDropGraph()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/parse/
Dresolve.cc167 param_ptr->set_func_graph(top_graph); in ConvertLoadedGraph()
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dcontrol_flow_pass.cc212 cur_node->set_func_graph(*after_fg); in CreateAfterGraph()
/third_party/mindspore/mindspore/lite/tools/anf_exporter/
Danf_exporter.cc884 cnode->set_func_graph(fg); in CreateCallCnode()
/third_party/mindspore/mindspore/ccsrc/pipeline/pynative/
Dpynative_execute.cc2556 resource->set_func_graph(bprop_graph); in GradNetInner()
2707 resource->set_func_graph(bprop_graph); in GetBpropGraph()