Skip to content
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

[Onnx] Pow support for other types #8933

Merged
merged 3 commits into from
Sep 8, 2021

Conversation

AndrewZhaoLuo
Copy link
Contributor

This updates the onnx spec for pow up to the latest opset. This includes mixing types and having integer pow().

We handle all of this by casting to float since codegen does not support an intrinsic pow operator. It might be worth it to come up with a good default implementation in relay for integer pow but I do not know if this is going to be faster than simply casting.

@masahi masahi merged commit 9a47fc0 into apache:main Sep 8, 2021
ylc pushed a commit to ylc/tvm that referenced this pull request Sep 29, 2021
* update pow

* update pow

* remove duplicate

Co-authored-by: Andrew Zhao Luo <andrewzhaoluo@system76-pc.localdomain>
ylc pushed a commit to ylc/tvm that referenced this pull request Jan 13, 2022
* update pow

* update pow

* remove duplicate

Co-authored-by: Andrew Zhao Luo <andrewzhaoluo@system76-pc.localdomain>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants