Skip to content

Commit b7efca1

Browse files
committed
not all transpose node has attr "perm"
1 parent 13d465b commit b7efca1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tf2onnx/optimizer/transpose_optimizer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,7 @@ def merge_duplicated_transposes(self):
115115
graph = self._g
116116
input_transposes_map = defaultdict(list)
117117
for node in graph.get_nodes():
118-
if node.type == "Transpose":
118+
if node.type == "Transpose" and node.get_attr("perm"):
119119
key = (node.input[0], str(node.get_attr("perm").ints))
120120
input_transposes_map[key].append(node)
121121

0 commit comments

Comments
 (0)