Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/lite/tools/converter/import/
Dmindir_control_flow_adjust.cc239 auto call_cnode = node->cast<CNodePtr>(); in InsertPartialFusionForRawCall() local
241 auto call_cnode_inputs = call_cnode->inputs(); in InsertPartialFusionForRawCall()
242 auto cnode_first_input = call_cnode->input(0); in InsertPartialFusionForRawCall()
254 partial_cnode->set_fullname_with_scope("partial_" + call_cnode->fullname_with_scope()); in InsertPartialFusionForRawCall()
255 call_cnode->set_inputs({partial_cnode}); in InsertPartialFusionForRawCall()
/third_party/mindspore/mindspore/lite/tools/common/
Dfunc_graph_subgraph.cc458 auto call_cnode = belong_anf_->NewCNode(call_node_inputs); in CreatePartialInBelongAnf() local
459 MS_CHECK_TRUE_MSG(call_cnode != nullptr, RET_NULL_PTR, "call_cnode is nullptr"); in CreatePartialInBelongAnf()
460 call_cnode->set_fullname_with_scope(graph_name + "/call"); in CreatePartialInBelongAnf()
466 manager->Replace(ori_output, call_cnode); in CreatePartialInBelongAnf()
606 auto call_cnode = utils::cast<CNodePtr>(call_node); in ApplySubGraph() local
607 MS_ASSERT(call_cnode != nullptr && call_cnode->inputs().size() == 1); in ApplySubGraph()
608 auto after_partial_node = call_cnode->inputs().at(0); in ApplySubGraph()
613 subgraph_nodes.insert(call_cnode); in ApplySubGraph()
/third_party/mindspore/mindspore/lite/tools/anf_exporter/
Danf_exporter.cc362 auto call_cnode = CreateCallCnode(fg, partial_cnode); in InsertCallNode() local
364 it = cnodes.insert(it, call_cnode); in InsertCallNode()
506 CNodePtr call_cnode = nullptr; in GetFinalGraph() local
509 call_cnode = fg_output->cast<CNodePtr>(); in GetFinalGraph()
515 auto cnode = call_cnode->input(kFirstDataIndex)->cast<CNodePtr>(); in GetFinalGraph()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/parse/
Dparse.cc661 …auto call_cnode = GenerateAnfNodeForCall(block, call_function_node, packed_arguments, group_argume… in ParseCall() local
663 call_cnode->set_interpret(true); in ParseCall()
665 return call_cnode; in ParseCall()