Compare node.attributes in is_same_node_merge (#178)

microsoft · Apr 9, 2021 · 43bc9da · 43bc9da
1 parent 91ef4b3
commit 43bc9da
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/onnxconverter_common/optimizer.py b/onnxconverter_common/optimizer.py
@@ -1611,6 +1611,9 @@ def is_same_node_merge(node_0, node_1, node):
         if node_0.origin.attribute != node_1.origin.attribute:
             return False
 
+        if node_0.attributes != node_1.attributes:
+            return False
+
         if len(node_0.origin.input) != len(node_1.origin.input):
             return False