-
Notifications
You must be signed in to change notification settings - Fork 3.4k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[Relay][OP] Fix bias_add default axis #2829
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM. Please fix the test cases :-)
@junrushao1994 I'm investigating whether we should go with this change... I realize that bias_add with default axis=1 works good with conv, but requires batch flatten for dense op. I'll follow up on this pr in a few days. |
any update? |
I decide to change the MXNet converter to explicit set axis=-1 in the bias_add. |
It looks good. Let's wait for the CI to complete |
@junrushao1994 @jroesch CI has passed. Could you help review the pr? |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
* Fix bias add default axis * update * Fix canonicalize ops for bias_add
* Fix bias add default axis * update * Fix canonicalize ops for bias_add
* Fix bias add default axis * update * Fix canonicalize ops for bias_add
* Fix bias add default axis * update * Fix canonicalize ops for bias_add
* Fix bias add default axis * update * Fix canonicalize ops for bias_add
* Fix bias add default axis * update * Fix canonicalize ops for bias_add
Change the default axis in bias_add from 1 to -1