We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 13d465b commit b7efca1Copy full SHA for b7efca1
tf2onnx/optimizer/transpose_optimizer.py
@@ -115,7 +115,7 @@ def merge_duplicated_transposes(self):
115
graph = self._g
116
input_transposes_map = defaultdict(list)
117
for node in graph.get_nodes():
118
- if node.type == "Transpose":
+ if node.type == "Transpose" and node.get_attr("perm"):
119
key = (node.input[0], str(node.get_attr("perm").ints))
120
input_transposes_map[key].append(node)
121
0 commit comments