Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/lite/tools/converter/parser/
Dunused_node_remove_pass.cc25 auto return_node = func_graph->get_return(); in ProcessGraph() local
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dinfershape_pass.cc302 auto return_node = sub_graph->get_return(); in SetSubGraphOutput() local
334 auto return_node = sub_graph->get_return(); in SetSubGraphAbstract() local
Dadd_tensor_array.cc52 auto return_node = func_graph->get_return(); in SetGraphOutput() local
Ddecrease_transpose_algo.cc551 auto return_node = sub_graph->get_return(); in SetSubGraphOutput() local
582 auto return_node = sub_graph->get_return(); in SetSubGraphAbstract() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/
Dfusion_base_pass.cc93 auto return_node = kernel_graph.get_return(); in MatchUBFusionPattern() local
/third_party/mindspore/mindspore/lite/tools/converter/import/
Dmindspore_importer.cc124 STATUS MindsporeImporter::GetFuncGraphOutputName(const CNodePtr &return_node) { in GetFuncGraphOutputName()
/third_party/mindspore/mindspore/lite/tools/common/
Dfunc_graph_subgraph.cc462 auto return_node = belong_anf_->get_return(); in CreatePartialInBelongAnf() local
602 auto return_node = belong_anf_->get_return(); in ApplySubGraph() local
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dtflite_lstm_cell_fusion.cc275 VectorRef return_node = VectorRef({is_return, make_tuple_node}); in GetBodyGraphPattern() local
315 auto return_node = func_graph->get_return(); in MatchGraph() local
Dtf_lstm_cell_fusion.cc177 VectorRef return_node = VectorRef({is_return, make_tuple_node}); in GetBodyGraphPattern() local
Dtf_bidirection_gru_fusion.cc370 VectorRef return_node = VectorRef({is_return, make_tuple_node}); in GetBodyGraphPattern() local
/third_party/mindspore/mindspore/core/load_mindir/
Danf_model_parser.cc995 auto return_node = outputFuncGraph->NewCNode(inputs); in BuildReturnForFuncGraph() local
1010 auto return_node = outputFuncGraph->NewCNode(inputs); in BuildReturnForFuncGraph() local
/third_party/glslang/glslang/MachineIndependent/
DpropagateNoContraction.cpp650 void propagateNoContractionInReturnNode(glslang::TIntermBranch* return_node) in propagateNoContractionInReturnNode()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/parse/
Dfunction_block.cc534 auto return_node = func_graph_->get_return(); in AttachIsolatedNodesBeforeReturn() local
Dparse.cc482 CNodePtr return_node = block_fg->NewCNodeInOrder({return_value_node, return_expr_node}); in ParseReturn() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/
Donnx_model_parser.cc447 auto return_node = loop_body_graph->get_return(); in BuildBodyGraph() local
762 auto return_node = subgraph->get_return(); in ConvertIfSubgraph() local
/third_party/mindspore/mindspore/lite/tools/anf_exporter/
Danf_exporter.cc280 schema::CNodeT *return_node) { in SetSubGraphOutputIndex()
/third_party/mindspore/mindspore/core/ir/
Dfunc_graph_cloner.cc193 auto return_node = iter->second->cast<CNodePtr>(); in CloneFuncGraphValueNodes() local
Dmanager.cc342 auto return_node = func_graph->get_return(); in AddFuncGraph() local
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dsession_basic.cc486 auto return_node = node->cast<CNodePtr>(); in SetReturnNode() local
766 auto return_node = kernel_graph->NewCNode({primitive, parameter}); in CreateSwitchInput() local
855 auto return_node = kernel_graph->NewCNode(return_inputs); in ProcessNodeRetFunc() local
Danf_runtime_algorithm.cc200 auto return_node = kg->get_return(); in KeepOrder() local
2275 auto return_node = root_graph->get_return(); in InsertMakeTupleForOutput() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dgraph_kernel_splitter.cc284 auto return_node = func_graph->NewCNode(return_inputs); in CreateReturnNode() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/cache_embedding/
Dcache_embedding.cc663 CNodePtr return_node; in CacheEmbeddingForTrain() local
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dgraph_scheduler.cc765 const auto &return_node = func_graph_to_call_num.first->get_return(); in BuildSwitchActor() local
807 const auto &return_node = func_graph->get_return(); in BuildGatherActor() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_parallel.cc2250 CNodePtr return_node = func_graph->get_return(); in FindLossCNode() local
2864 auto return_node = func_graph->get_return(); in MarkForwardCNode() local