Browse Source

Modify to GetOutNodes

tags/v1.3.0
zhangxiaokun 4 years ago
parent
commit
8a425b7335
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/graph/passes/base_pass.cc

+ 1
- 1
ge/graph/passes/base_pass.cc View File

@@ -259,7 +259,7 @@ Status GEPass::RunPassesOneGraph(const NamesToPass &names_to_passes) {
} }


std::vector<NodePtr> nodes_to_pass; std::vector<NodePtr> nodes_to_pass;
GetNextIterNodes(node->GetOutAllNodes(), nodes_to_pass, during_pass_node_set);
GetNextIterNodes(node->GetOutNodes(), nodes_to_pass, during_pass_node_set);


auto ret = RunPasses(node, names_to_passes, during_pass_node_set); auto ret = RunPasses(node, names_to_passes, during_pass_node_set);
if (ret != SUCCESS) { if (ret != SUCCESS) {


Loading…
Cancel
Save