|
|
|
@ -22,6 +22,8 @@
|
|
|
|
|
#include "graph/preprocess/multi_batch_options.h"
|
|
|
|
|
#include "graph/utils/node_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"
|
|
|
|
|
|
|
|
|
|
namespace ge {
|
|
|
|
@ -614,7 +616,7 @@ Status MultiBatchClonePass::CreateSubgraphs(const ComputeGraphPtr &graph, const
|
|
|
|
|
graph->AddSubgraph(subgraph->GetName(), subgraph);
|
|
|
|
|
all_branch_output_[subgraph] = subgraph->FindFirstNodeMatchType(NETOUTPUT);
|
|
|
|
|
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);
|
|
|
|
|
op_desc->AddSubgraphName(key_name);
|
|
|
|
|