-
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][DYN] Implementation of the dynamic pad operator #6284
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.
This appears to have #6273 in it as well. Can you separate the upsampling work?
@zhiics @icemelon9 @mbrookhart had a few bugs on this, but it's good to go now! Please let me know what you think. Thanks |
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.
Nitpick, LGTM, thanks!
python/tvm/relay/op/nn/nn.py
Outdated
@@ -1147,13 +1149,13 @@ def upsampling(data, | |||
|
|||
Parameters | |||
---------- | |||
data : tvm.relay.Expr | |||
data : tvm.relay.Expr or tuple<anytype> or list<anytype> |
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.
I don't think data can be a tuple or list?
Thanks @mbrookhart @electriclilies ! |
Implementation of the dynamic pad operator
@mbrookhart @zhiics please take a look and let me know if you have any suggestions
Also, clang-format changed a bunch of stuff in include/tvm/relay/attrs/nn.h -- what files are supposed to be formatted using clang-format?