| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/ |
| D | graph_converter.cc | 97 const CNodePtr &make_tuple_cnode = root->NewCNode(make_tuple_inputs); in BuildMakeTupleNode() local
98 MS_EXCEPTION_IF_NULL(make_tuple_cnode); in BuildMakeTupleNode()
99 AnfAlgo::SetOutputInferTypeAndShape(make_tuple_types, make_tuple_shapes, make_tuple_cnode.get()); in BuildMakeTupleNode()
101 return make_tuple_cnode; in BuildMakeTupleNode()
214 const CNodePtr &make_tuple_cnode = BuildMakeTupleNode(root, anf_trt_index_map, trt_node); in ReplaceSubgraphWithTrtNode() local
215 manager->Replace(outputs[0], make_tuple_cnode); in ReplaceSubgraphWithTrtNode()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | updatestate_eliminate.cc | 481 auto make_tuple_cnode = make_tuple->cast<CNodePtr>(); in EliminateUpdateStateBetweenAssignMakeTuple() local
482 if (make_tuple_cnode->size() != kMakeTupleSize) { in EliminateUpdateStateBetweenAssignMakeTuple()
485 auto a1 = make_tuple_cnode->input(kInputIndex); in EliminateUpdateStateBetweenAssignMakeTuple()
486 auto a2 = make_tuple_cnode->input(kAttachIndex); in EliminateUpdateStateBetweenAssignMakeTuple()
510 …AnfNodePtrList new_make_tuple_inputs{NewValueNode(prim::kPrimMakeTuple), make_tuple_cnode->input(k… in EliminateUpdateStateBetweenAssignMakeTuple()
511 make_tuple_cnode->input(kAttachIndex), assign}; in EliminateUpdateStateBetweenAssignMakeTuple()
540 auto make_tuple_cnode = make_tuple->cast<CNodePtr>(); in EliminateUpdateStateBetweenMakeTupleAssign() local
541 if (make_tuple_cnode->size() != kMakeTupleSize || !OnlyUsedByOneNode(make_tuple, update_state)) { in EliminateUpdateStateBetweenMakeTupleAssign()
544 auto a2 = make_tuple_cnode->input(kInputIndex); in EliminateUpdateStateBetweenMakeTupleAssign()
545 auto a3 = make_tuple_cnode->input(kAttachIndex); in EliminateUpdateStateBetweenMakeTupleAssign()
[all …]
|
| D | partial_eliminate.h | 391 auto make_tuple_cnode = switch_layer_cnode->input(2)->cast<CNodePtr>(); in operator() local
392 if (make_tuple_cnode->size() < 2) { in operator()
399 for (size_t i = 1; i < make_tuple_cnode->size(); ++i) { in operator()
400 Visit(make_tuple_cnode->input(i)); in operator()
430 auto make_tuple_cnode = switch_layer_cnode->input(2)->cast<CNodePtr>(); in BuildNewSwitchLayerNode() local
431 AnfNodePtrList make_tuple_args{make_tuple_cnode->input(0)}; in BuildNewSwitchLayerNode()
433 TraceGuard guard1(std::make_shared<TraceCopy>(make_tuple_cnode->debug_info())); in BuildNewSwitchLayerNode()
|
| /third_party/mindspore/mindspore/lite/tools/converter/acl/mapper/ |
| D | spatial_node_adapter.cc | 121 auto make_tuple_cnode = func_graph->NewCNode(new_inputs); in AdapteNodeWithDynamicInput() local
122 if (make_tuple_cnode == nullptr) { in AdapteNodeWithDynamicInput()
127 const std::vector<AnfNodePtr> replace_node = {cnode_inputs[0], make_tuple_cnode}; in AdapteNodeWithDynamicInput()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/graph/ |
| D | add_tensor_array.cc | 88 …auto make_tuple_cnode = func_graph->NewCNode({make_tuple_vnode, output_node, tensor_array_write_no… in SetGraphOutput() local
89 if (make_tuple_cnode == nullptr) { in SetGraphOutput()
93 make_tuple_cnode->set_fullname_with_scope("return tuple"); in SetGraphOutput()
103 auto new_return_node = func_graph->NewCNode({return_value_node, make_tuple_cnode}); in SetGraphOutput()
|
| /third_party/mindspore/mindspore/lite/tools/converter/import/ |
| D | mindir_control_flow_adjust.cc | 138 auto make_tuple_cnode = after_fg->NewCNode(make_tuple_inputs); in AddAfterFuncGraph() local
139 MS_CHECK_TRUE_MSG(make_tuple_cnode != nullptr, nullptr, "Failed to create C node."); in AddAfterFuncGraph()
140 make_tuple_cnode->set_fullname_with_scope("return tuple"); in AddAfterFuncGraph()
148 std::vector<AnfNodePtr> op_inputs = {value_node, make_tuple_cnode}; in AddAfterFuncGraph()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tf/ |
| D | functionalize_while.cc | 559 auto make_tuple_cnode = body_sub_func_graph_->NewCNode(make_tuple_inputs); in IdentifyBodySubgraphOutput() local
560 CHECK_NULL_RETURN(make_tuple_cnode); in IdentifyBodySubgraphOutput()
561 make_tuple_cnode->set_fullname_with_scope(return_cnode->fullname_with_scope() + "tuple"); in IdentifyBodySubgraphOutput()
563 return_cnode->add_input(make_tuple_cnode); in IdentifyBodySubgraphOutput()
|
| D | tf_model_parser.cc | 1155 auto make_tuple_cnode = anf_graph->NewCNode(make_tuple_inputs); in MakeAnfGraphOutputs() local
1156 CHECK_NULL_RETURN(make_tuple_cnode); in MakeAnfGraphOutputs()
1157 make_tuple_cnode->set_fullname_with_scope("return_tuple"); in MakeAnfGraphOutputs()
1166 std::vector<AnfNodePtr> op_inputs = {value_node, make_tuple_cnode}; in MakeAnfGraphOutputs()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/caffe/ |
| D | caffe_model_parser.cc | 434 auto make_tuple_cnode = res_graph_->NewCNode(make_tuple_inputs); in ConvertGraphOutputs() local
435 MSLITE_CHECK_PTR(make_tuple_cnode); in ConvertGraphOutputs()
436 make_tuple_cnode->set_fullname_with_scope("return tuple"); in ConvertGraphOutputs()
444 op_inputs.emplace_back(make_tuple_cnode); in ConvertGraphOutputs()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/ |
| D | tflite_model_parser.cc | 485 auto make_tuple_cnode = func_graph->NewCNode(make_tuple_inputs); in ConvertGraphOutputs() local
486 MSLITE_CHECK_PTR(make_tuple_cnode); in ConvertGraphOutputs()
487 make_tuple_cnode->set_fullname_with_scope("return_tuple"); in ConvertGraphOutputs()
496 op_inputs.emplace_back(make_tuple_cnode); in ConvertGraphOutputs()
|
| /third_party/mindspore/mindspore/lite/tools/common/ |
| D | graph_util.cc | 52 auto make_tuple_cnode = graph->NewCNode(make_tuple_prim_ptr, outputs); in SetFuncGraphOutput() local
57 make_tuple_cnode->set_fullname_with_scope("return tuple"); in SetFuncGraphOutput()
58 graph->set_output(make_tuple_cnode, false); in SetFuncGraphOutput()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/ |
| D | onnx_model_parser.cc | 345 auto make_tuple_cnode = anf_graph->NewCNode(make_tuple_prim_ptr, make_tuple_inputs); in ConvertGraphOutputs() local
346 if (make_tuple_cnode == nullptr) { in ConvertGraphOutputs()
350 make_tuple_cnode->set_fullname_with_scope("return tuple"); in ConvertGraphOutputs()
351 return_inputs.emplace_back(make_tuple_cnode); in ConvertGraphOutputs()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/ |
| D | pipeline_transformer.cc | 530 auto make_tuple_cnode = output->cast<CNodePtr>(); in FindPipelineCareNode() local
531 output = make_tuple_cnode->input(LongToSize(get_item_index + 1)); in FindPipelineCareNode()
|