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

[Relay][Frontend][Onnx] SequenceAt and SplitToSequence Operators #13602

Merged
merged 4 commits into from
Dec 13, 2022

Conversation

jwfromm
Copy link
Contributor

@jwfromm jwfromm commented Dec 12, 2022

This PR adds support for some new sequence operators in the onnx frontend. I made a few other small tweaks to get everything working, specifically an attempt to rectify dynamic rank in If statements and the removal of a superfluous check in concat's shape function. With these changes, we can import and run the latest Yolo variants which is pretty cool.

@tvm-bot
Copy link
Collaborator

tvm-bot commented Dec 12, 2022

Thanks for contributing to TVM! Please refer to the contributing guidelines https://tvm.apache.org/docs/contribute/ for useful information and tips. Please request code reviews from Reviewers by @-ing them in a comment.

Generated by tvm-bot

@jwfromm
Copy link
Contributor Author

jwfromm commented Dec 13, 2022

@AndrewZhaoLuo @michalpiszczek @sfvaroglu can you guys take a look at this PR?

python/tvm/relay/frontend/onnx.py Outdated Show resolved Hide resolved
python/tvm/relay/frontend/onnx.py Outdated Show resolved Hide resolved
@sfvaroglu
Copy link
Contributor

LGTM! Thanks @jwfromm!

@AndrewZhaoLuo AndrewZhaoLuo merged commit c547bbb into apache:main Dec 13, 2022
fzi-peccia pushed a commit to fzi-peccia/tvm that referenced this pull request Mar 27, 2023
…che#13602)

* Add support for SequenceAt and SplitToSequence to onnx importer

* Formatting

* Change keepdims comparison

* Only unify non-tuples in If
@jwfromm jwfromm deleted the sequence_at_split branch April 12, 2023 15:57
mikeseven pushed a commit to mikeseven/tvm that referenced this pull request Sep 27, 2023
…che#13602)

* Add support for SequenceAt and SplitToSequence to onnx importer

* Formatting

* Change keepdims comparison

* Only unify non-tuples in If
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.

4 participants