| @@ -22,6 +22,8 @@ | |||||
| #include "graph/preprocess/multi_batch_options.h" | #include "graph/preprocess/multi_batch_options.h" | ||||
| #include "graph/utils/node_utils.h" | #include "graph/utils/node_utils.h" | ||||
| #include "graph/utils/op_desc_utils.h" | #include "graph/utils/op_desc_utils.h" | ||||
| #include "graph/utils/tensor_utils.h" | |||||
| #include "graph/utils/type_utils.h" | |||||
| #include "register/op_registry.h" | #include "register/op_registry.h" | ||||
| namespace ge { | namespace ge { | ||||
| @@ -614,7 +616,7 @@ Status MultiBatchClonePass::CreateSubgraphs(const ComputeGraphPtr &graph, const | |||||
| graph->AddSubgraph(subgraph->GetName(), subgraph); | graph->AddSubgraph(subgraph->GetName(), subgraph); | ||||
| all_branch_output_[subgraph] = subgraph->FindFirstNodeMatchType(NETOUTPUT); | all_branch_output_[subgraph] = subgraph->FindFirstNodeMatchType(NETOUTPUT); | ||||
| GE_CHK_STATUS_RET(UpdateSubgraphOutput(all_branch_output_[subgraph]), | GE_CHK_STATUS_RET(UpdateSubgraphOutput(all_branch_output_[subgraph]), | ||||
| "Update %s failed", all_branch_output_[subgraph]->GetName().c_str()); | |||||
| "Update %s failed", all_branch_output_[subgraph]->GetName().c_str()); | |||||
| const string key_name = "branches" + std::to_string(i); | const string key_name = "branches" + std::to_string(i); | ||||
| op_desc->AddSubgraphName(key_name); | op_desc->AddSubgraphName(key_name); | ||||