-
Notifications
You must be signed in to change notification settings - Fork 3.5k
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] Support deformable Conv2D NHWC #7075
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.
there are some test cases in tests/python/relay/test_op_level5.py
Thanks for the pointer. Added 2 test cases to op level 5. I only added the infer type cases but not the execution because of the naive schedule on GPU. |
b88f4d6
to
e35b7de
Compare
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
* [Relay] Support deformable conv2D NHWC * add test case * fix lint * lint
* [Relay] Support deformable conv2D NHWC * add test case * fix lint * lint
* [Relay] Support deformable conv2D NHWC * add test case * fix lint * lint
PR #6999 added a TOPI compute for deformable Conv2D NHWC. This PR adds the required supports to lower NHWC deformable Conv2D from Relay so that it can be tuned by auto_scheduler.
Notes:
I didn't find a proper place to add unit tests. Please suggest the right place if we need them.cc @icemelon9 @jroesch @merrymercy @vinx13 @Laurawly