Home
last modified time | relevance | path

Searched defs:return_node (Results 1 – 25 of 51) sorted by relevance

123

/third_party/mindspore/mindspore-src/source/tests/ut/cpp/mindir/
Dtest_node_attr_export.cc57 auto return_node = dstgraph_ptr->output(); in TEST_F() local
84 auto return_node = graph->output(); in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/
Derase_not_cut_attr.cc24 AnfNodePtr return_node = func_graph->get_return(); in Run() local
Derase_invalid_micro_depend.cc47 AnfNodePtr return_node = graph->get_return(); in EraseMicroDepend() local
Dswitch_not_cut.cc185 AnfNodePtr return_node = func_graph->get_return(); in IsValidFuncGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/
Dcomm_op_attrs.cc33 AnfNodePtr return_node = graph->get_return(); in CommOpAttrs() local
Dprocess_send_recv_for_ge.cc152 AnfNodePtr return_node = graph->get_return(); in AddAllGatherRecvDepend() local
250 AnfNodePtr return_node = graph->get_return(); in AddSendClosureDepend() local
354 AnfNodePtr return_node = graph->get_return(); in ProcessSendRecvForGE() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/irpass/
Dpynative_no_grad_eliminate.h95 const auto &return_node = func_graph_->get_return(); in IsNeedOptimize() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/
Dunused_node_remove_pass.cc31 auto return_node = func_graph->get_return(); in ProcessGraph() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/runtime/graph_scheduler/
Dcontrol_node_parser_test.cc59 auto return_node = kernel_graph->NewCNode(return_inputs); in BuildKernelGraph() local
105 auto return_node = root_func_graph->NewCNode(return_inputs); in BuildGraphs() local
Dgraph_compiler_test.cc70 auto return_node = func_graph->NewCNode(return_inputs); in TEST_F() local
Dany_type_kernel_actor_test.cc51 auto return_node = func_graph->NewCNode(return_inputs); in BuildAnyTypeGraph() local
Dgraph_scheduler_test.cc192 auto return_node = func_graph->NewCNode(return_inputs); in BuildAnyTypeGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dspecify_graph_output_format.cc54 auto return_node = graph->get_return(); in HandleGraphOutput() local
Dinput_data_type_trans_pass.cc101 auto return_node = graph->get_return(); in HandleGraphOutput() local
Doutput_variable_pass.cc40 auto return_node = graph->get_return(); in Run() local
Dcore_infershape_pass.cc312 auto return_node = sub_graph->get_return(); in SetSubGraphOutput() local
342 auto return_node = sub_graph->get_return(); in SetSubGraphAbstract() local
Dinfershape_pass.cc393 auto return_node = sub_graph->get_return(); in SetSubGraphOutput() local
425 auto return_node = sub_graph->get_return(); in SetSubGraphAbstract() local
Dadd_tensor_array.cc56 auto return_node = func_graph->get_return(); in SetGraphOutput() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/cache_embedding/
Dps_embedding_cache_inserter.cc272 auto return_node = graph->NewCNode(return_inputs); in CreateReturnNode() local
317 CNodePtr return_node = CreateReturnNode(graph, send_node); in ConstructEmbeddingLookupSubGraph() local
369 CNodePtr return_node = CreateReturnNode(graph, embedding_cache_update_node); in ConstructUpdateEmbeddingSubGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dreduce_same_op_in_horizon.cc99 auto return_node = func_graph->get_return(); in Process() local
Dtflite_lstm_cell_fusion.cc285 VectorRef return_node = VectorRef({is_return, make_tuple_node}); in GetBodyGraphPattern() local
325 auto return_node = func_graph->get_return(); in MatchGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/optimizer/dynamic_shape/
Dlink_custom_op.cc170 auto return_node = g->get_return(); in AttachDependNodes() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Dadd_noop_to_es_grad.cc99 auto return_node = func_graph->get_return(); in InsertNoOpForOutput() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/common/
Dfunc_graph_subgraph.cc466 auto return_node = belong_anf_->get_return(); in CreatePartialInBelongAnf() local
606 auto return_node = belong_anf_->get_return(); in ApplySubGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/
Dfold_updatestate.cc80 const auto &return_node = func_graph->get_return(); in GetTailDepend() local

123