fix error fusion in transop breadth fusion pass

5 years ago · 70b2a12c81
parent d5e6e44a65
commit 70b2a12c81
1 changed files with 3 additions and 1 deletions
--- a/ge/graph/passes/transop_breadth_fusion_pass.cc
+++ b/ge/graph/passes/transop_breadth_fusion_pass.cc
@ -70,8 +70,10 @@ std::string TransOpBreadthFusionPass::GetNodeId(const int anchor_index, const No
    trans_data_type = true;
    trans_format = true;
    trans_shape = true;
-  } else if (node->GetType() == RESHAPE) {
+  } else if (node->GetType() == RESHAPE || node->GetType() == EXPANDDIMS || node->GetType() == SQUEEZE) {
    trans_shape = true;
+  } else if (node->GetType() == REFORMAT) {
+    trans_format = true;
  }

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