Browse Source

fix error fusion in transop breadth fusion pass

tags/v1.2.0
lichun 3 years ago
parent
commit
70b2a12c81
1 changed files with 3 additions and 1 deletions
  1. +3
    -1
      ge/graph/passes/transop_breadth_fusion_pass.cc

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

@@ -70,8 +70,10 @@ std::string TransOpBreadthFusionPass::GetNodeId(const int anchor_index, const No
trans_data_type = true; trans_data_type = true;
trans_format = true; trans_format = true;
trans_shape = true; trans_shape = true;
} else if (node->GetType() == RESHAPE) {
} else if (node->GetType() == RESHAPE || node->GetType() == EXPANDDIMS || node->GetType() == SQUEEZE) {
trans_shape = true; trans_shape = true;
} else if (node->GetType() == REFORMAT) {
trans_format = true;
} }


id << node->GetType() << '-' << anchor_index; id << node->GetType() << '-' << anchor_index;


Loading…
Cancel
Save