Skip to content
Snippets Groups Projects
Commit 99d81dfc authored by Yaman Umuroglu's avatar Yaman Umuroglu
Browse files

[Streamline] bugfix in AbsorbConsecutiveTransposes

without this, a single trailing Transpose at the end of the NN
gets eaten up
parent 389a9156
No related branches found
No related tags found
No related merge requests found
......@@ -492,6 +492,8 @@ class AbsorbConsecutiveTransposes(Transformation):
if node.op_type == "Transpose":
next_nodes = model.find_consumers(node.output[0])
perms1 = list(get_by_name(node.attribute, "perm").ints)
if len(next_nodes) == 0:
continue
# check if all nodes after fork are opposite transposes
all_opposite_transposes = True
for next_node in next_nodes:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment