-
Notifications
You must be signed in to change notification settings - Fork 3.8k
Relay reshape reshape_like compute and schedule #2159
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 reshape reshape_like compute and schedule #2159
Conversation
|
@jroesch, @MarisaKirisame @tqchen @yzhliu Please help to review this PR. Thanks |
|
|
||
| func = relay.Function([x, y], z) | ||
|
|
||
| for target, ctx in ctx_list(): |
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.
this code appear in other test as well. should we refactor it to a common facility?
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.
👍
|
Looks good to me, could use this for #2144, I copied the compute to my branch for the time being. |
|
The reshape implementation has some issues, due to the possibility of newshape containing negative number, we need to use the type of output instead to implement this |
|
Given the interest of recent changes, I will merge this in and amend in the followup PR #2163. Thanks @siju-samuel @MarisaKirisame @jroesch |
|
please do refer to the #2163 for the necessary changes and use that as reference for future codes :) |
Repair imports First test passes Fix bug in axes passing Add implementations for upsampling and pad Add support for bias with conv ops Rebase fixup Borrow the reshape changes while waiting on apache#2159 Debugging tests Repair after rebase Refactor to use common functionality Fix helper func Remove final eval Remove some debuggin Remove unecessary ws change Fix multiple outputs Tests pass Enable all tests Rebase repair Fix last test case Fix last test case All tests but RNNs pass Fix tests Remove final eval Reformat Fix bugs in MxNet Fix 3.5 support in Relay, and rename Factor common functionality into one file Apply most of code review Fix linting Fix NNVM linting Fix test error Fix MLP test Fix linting error One more linting issue Clean up diff and fix bugs Retrigger disable MLP test Retrigger Retrigger Roll back NNVM change Retrigger Retrigger Retrigger
Repair imports First test passes Fix bug in axes passing Add implementations for upsampling and pad Add support for bias with conv ops Rebase fixup Borrow the reshape changes while waiting on apache#2159 Debugging tests Repair after rebase Refactor to use common functionality Fix helper func Remove final eval Remove some debuggin Remove unecessary ws change Fix multiple outputs Tests pass Enable all tests Rebase repair Fix last test case Fix last test case All tests but RNNs pass Fix tests Remove final eval Reformat Fix bugs in MxNet Fix 3.5 support in Relay, and rename Factor common functionality into one file Apply most of code review Fix linting Fix NNVM linting Fix test error Fix MLP test Fix linting error One more linting issue Clean up diff and fix bugs Retrigger disable MLP test Retrigger Retrigger Roll back NNVM change Retrigger Retrigger Retrigger
#2051
Thanks for contributing to TVM! Please refer to guideline https://docs.tvm.ai/contribute/ for useful information and tips. After the pull request is submitted, please request code reviews from Reviewers.