-
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][Frontend] TF Tile Round Sign Pow Exp Reverse #2960
Conversation
pls ignore CI, failed due to "raise RemoteDisconnected" in mxnet test case, not related to my change |
@srkreddy1238 @kazum @jroesch @zhiics @wweic @lixiaoquan ci passed, please help review |
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. thanks for reordering the operator map, it might be good to mention which operators are newly added.
@wweic thanks. The newly added list: Tile, ReverseV2, Round, Sign, Pow, Exp |
compare_tf_with_tvm([np_data], ['in_data:0'], 'reverse:0') | ||
|
||
def test_forward_reverse_v2(): | ||
"""test ReverseV2""" |
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.
Add a negative axis if supported by TVM.
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.
Added, yeah tvm supports this case.
@srkreddy1238 please followup and manage this PR |
Thanks @yongwww . This is now merged. |
* [Relay][Frontend] TF Round Sign Pow Exp Reverse * fix ci * fix comments
* [Relay][Frontend] TF Round Sign Pow Exp Reverse * fix ci * fix comments
Add Tile, ReverseV2, Round, Sign, Pow, Exp in tf converter, most of them are needed for object detection models like ssd.
@srkreddy1238 @kazum @jroesch @zhiics @wweic pls help review