Browse Source

fix

tags/v1.2.0
wjm 4 years ago
parent
commit
37e937aedc
1 changed files with 3 additions and 1 deletions
  1. +3
    -1
      ge/graph/passes/multi_batch_clone_pass.cc

+ 3
- 1
ge/graph/passes/multi_batch_clone_pass.cc View File

@@ -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);


Loading…
Cancel
Save