Skip to content

Conversation

@zhijxu-MS
Copy link
Collaborator

fix transpose optimizer bug

node can't be removed directly when graph output comes from it

@zhijxu-MS zhijxu-MS changed the title Tmp branch for pr3 fix some transpose optimizer bugs Mar 26, 2019
@nbcsm nbcsm requested a review from pengwa March 26, 2019 07:32
self._g.replace_all_inputs(ops, node.output[0], trans.input[0])

shape = self._g.get_shape(node.output[0])
dtype = self._g.get_dtype(node.output[0])
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

please add tests to cover these cases.

Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

1 transpose node is graph output has been covered by test_transpose_leaky_relu
2 identity is graph output is covered by new-added test case "test_transpose_with_identity"

mul input may come from same transpose node
@zhijxu-MS zhijxu-MS force-pushed the tmp_branch_for_PR3 branch from ef4a834 to 2594eca Compare March 27, 2019 06:36
it's ok in local env, while in unit test the target info is not given so
disable them for now
@nbcsm nbcsm merged commit eb31fc3 into onnx:master Mar 27, 2019
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants