Fix naming transpose nodes

microsoft · Apr 29, 2024 · a220b1b · a220b1b
1 parent e55b976
commit a220b1b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/python/py/models/builder.py b/src/python/py/models/builder.py
@@ -528,7 +528,7 @@ def make_mul(self, name, inputs, dtype, shape):
 
     def make_transpose(self, name, root_input, dtype, shape, perm):
         output = f"{name}/output_0"
-        self.make_node("Transpose", inputs=[root_input], outputs=[output], perm=perm)
+        self.make_node("Transpose", inputs=[root_input], outputs=[output], name=name, perm=perm)
         self.make_value_info(output, dtype, shape=shape)
 
     def make_matmul(self, matmul, name, root_input, **kwargs):