Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dgraph_kernel_expander.cc166 AnfNodePtr PyExpander::CreateExpandGraphKernel(const FuncGraphPtr &new_func_graph, const CNodePtr &… in CreateExpandGraphKernel() argument
171 EliminateRedundantParameters(new_func_graph, &inputs); in CreateExpandGraphKernel()
172 kernel::GetValidKernelNodes(new_func_graph, &kernel_nodes); in CreateExpandGraphKernel()
173 kernel::GetFuncGraphOutputNodes(new_func_graph, &outputs); in CreateExpandGraphKernel()
174 auto graph_kernel_node = CreateNewFuseCNode(func_graph, new_func_graph, inputs, outputs); in CreateExpandGraphKernel()
175 SetNewKernelInfo(graph_kernel_node, new_func_graph, inputs, outputs); in CreateExpandGraphKernel()
184 auto new_func_graph = CreateExpandFuncGraph(cnode); in Run() local
185 if (new_func_graph == nullptr) { in Run()
188 new_func_graph->set_attr(FUNC_GRAPH_ATTR_GRAPH_KERNEL, MakeValue(AnfAlgo::GetCNodeName(cnode))); in Run()
189 auto graph_kernel_node = CreateExpandGraphKernel(new_func_graph, cnode); in Run()
Dgraph_kernel_expander.h39 …virtual AnfNodePtr CreateExpandGraphKernel(const FuncGraphPtr &new_func_graph, const CNodePtr &old…
/third_party/mindspore/mindspore/core/ir/
Dfunc_graph_cloner.cc737 auto new_func_graph = std::make_shared<FuncGraph>(); in TransformableClone() local
740 …(void)std::for_each(parameters.begin(), parameters.end(), [&new_func_graph](const AnfNodePtr &para… in TransformableClone()
743 (void)new_func_graph->add_parameter()->set_abstract(param->abstract()); in TransformableClone()
747 cloner.AddClone(func_graph, new_func_graph, new_func_graph->parameters()); in TransformableClone()
749 new_func_graph->set_output(output); in TransformableClone()
750 new_func_graph->set_has_vararg(func_graph->has_vararg()); in TransformableClone()
751 new_func_graph->set_has_kwarg(func_graph->has_kwarg()); in TransformableClone()
752 new_func_graph->set_kwonlyargs_count(func_graph->kwonlyargs_count()); in TransformableClone()
753 new_func_graph->set_hyper_param_count(func_graph->hyper_param_count()); in TransformableClone()
754 new_func_graph->set_is_generate(func_graph->is_generated()); in TransformableClone()
[all …]
/third_party/mindspore/mindspore/lite/tools/converter/quantizer/
Dquantize_util.cc586 auto new_func_graph = cloner[func_graph]; in CopyFuncGraph() local
593 for (auto &cnode : new_func_graph->GetOrderedCnodes()) { in CopyFuncGraph()
628 return new_func_graph; in CopyFuncGraph()