Home
last modified time | relevance | path

Searched defs:return_cnode (Results 1 – 20 of 20) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/utils/
Dbuild_func_graph.cc58 auto return_cnode = graph->NewCNode(return_prim_c, {return_input}); in AddReturn() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dcore_infershape_pass.cc33 int JudgeControlFlowCertainOutputHasInferred(const CNodePtr &return_cnode, size_t index, bool *infe… in JudgeControlFlowCertainOutputHasInferred()
65 auto return_cnode = sub_graph->get_return(); in ModifyWhileBodyGraphInputs() local
79 int MergeTwoBranchOfIfOp(const CNodePtr &cnode, const CNodePtr &return_cnode, size_t index, bool *t… in MergeTwoBranchOfIfOp()
Dinfershape_pass.cc78 int JudgeControlFlowCertainOutputHasInferred(const CNodePtr &return_cnode, size_t index, bool *infe… in JudgeControlFlowCertainOutputHasInferred()
110 auto return_cnode = sub_graph->get_return(); in ModifyWhileBodyGraphInputs() local
140 int MergeTwoBranchOfIfOp(const CNodePtr &cnode, const CNodePtr &return_cnode, size_t index, bool *t… in MergeTwoBranchOfIfOp()
Dadd_tensor_array.cc61 auto return_cnode = return_node->cast<CNodePtr>(); in SetGraphOutput() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/tools/converter/registry/parser/
Dmodel_parser_test.cc161 auto return_cnode = res_graph_->NewCNode(return_prim, {nodes_[outputs[0]]}); in BuildGraphOutputs() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/common/
Dgraph_output_name_keeper.cc37 auto return_cnode = func_graph->get_return(); in SaveOriginalOutputs() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/tools/optimizer/fusion/fusion_inout_test/
Dfusion_inout_test.cc121 auto return_cnode = graph->NewCNode(return_prim_c, {return_input}); in AddReturn() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/tf/
Dfunctionalize_while.cc394 auto return_cnode = cond_sub_func_graph_->NewCNode(op_inputs); in IdentifyCondSubgraphOutput() local
558 auto return_cnode = body_sub_func_graph_->NewCNode(op_inputs); in IdentifyBodySubgraphOutput() local
Dfunctionalize_cond.cc203 auto return_cnode = graph->NewCNode(op_inputs); in CreateBranchGraph() local
Dtf_model_parser.cc1366 auto return_cnode = anf_graph->NewCNode(op_inputs); in MakeAnfGraphOutputs() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/import/
Dmindir_control_flow_adjust.cc173 auto return_cnode = after_fg->NewCNode(op_inputs); in AddAfterFuncGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/om/
Dom_model_parser.cc133 auto return_cnode = func_graph->NewCNode(return_prim_c, {output_node}); in CreateGraphOutputs() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/tools/converter/adapter/acl/mapper/
Dclip_mapper_test.cc82 auto return_cnode = graph->NewCNode(return_prim_c, {return_input}); in AddReturn() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/src/
Dcustom_creator.cc439 auto return_cnode = func_graph->get_return(); in SetCustomMultiOutput() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/pytorch/
Dpytorch_model_parser.cc200 auto return_cnode = anf_graph->NewCNode(return_prim, return_inputs); in BuildReturnNode() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/onnx/
Donnx_model_parser.cc237 auto return_cnode = anf_graph->NewCNode(return_prim_c, return_inputs); in BuildReturnNode() local
945 auto return_cnode = return_node->input(1)->cast<CNodePtr>(); in ConvertIfSubgraph() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/cxx_api/model/
Dmodel_impl.cc105 auto return_cnode = func_graph->NewCNode(return_prim_c, {custom_cnode}); in CreateFuncGraphFromDataFlow() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/
Dgraph_kernel_recompute.cc594 auto return_cnode = return_node->cast<CNodePtr>(); in FindNodeRecomputeCandidates() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/lite_exporter/
Danf_exporter.cc510 int AnfExporter::SetTailCallForReturn(const CNodePtr &return_cnode) { in SetTailCallForReturn()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/parse/
Dparse.cc1188 …CNodePtr return_cnode = func_graph->NewCNodeInOrder({NewValueNode(prim::kPrimReturn), return_expr_… in ParseReturn() local