Skip to content

[ONNX] Add support for GatherElements conversion - #6446

Merged
masahi merged 3 commits into
apache:masterfrom
masahi:torch-onnx-gatherelements
Sep 12, 2020
Merged

[ONNX] Add support for GatherElements conversion#6446
masahi merged 3 commits into
apache:masterfrom
masahi:torch-onnx-gatherelements

Conversation

@masahi

Copy link
Copy Markdown
Member

https://github.com/onnx/onnx/blob/master/docs/Operators.md#GatherElements

This is required to convert decision trees from hummingbird to Relay.

please review @siju-samuel@jwfromm@mbrookhart

@mbrookhartmbrookhart left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, thanks!

@siju-samuelsiju-samuel left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, just remove the prints

Comment threadtests/python/frontend/onnx/test_forward.py Outdated
@masahi
masahiforce-pushed the torch-onnx-gatherelements branch from f94b408 to 6cac63bCompareSeptember 11, 2020 03:02

@siju-samuelsiju-samuel left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM.

@masahi

Copy link
Copy Markdown
MemberAuthor

@siju-samuel please don't merge this PR until we first let #6448 in (I was asked by Jared to merge that PR as soon as possible before any other PR, to avoid conflict)

@masahi
masahiforce-pushed the torch-onnx-gatherelements branch from 6cac63b to f744383CompareSeptember 11, 2020 13:20
Comment threadpython/tvm/relay/frontend/onnx.py Outdated
@masahi
masahiforce-pushed the torch-onnx-gatherelements branch from f744383 to 459adfaCompareSeptember 11, 2020 20:31
@masahi

Copy link
Copy Markdown
MemberAuthor

@tqchen I've got "no space left on device" from CI. Is CI having a problem?
https://ci.tvm.ai/blue/organizations/jenkins/tvm/detail/PR-6446/4/pipeline

@masahi
masahiforce-pushed the torch-onnx-gatherelements branch from 459adfa to c76317eCompareSeptember 12, 2020 00:18
@mbrookhart

Copy link
Copy Markdown
Contributor

I got the same error on another PR

@tqchen

tqchen commented Sep 12, 2020

Copy link
Copy Markdown
Member

@masahi@mbrookhart please retrigger, the space issue has been resolved

@masahi
masahiforce-pushed the torch-onnx-gatherelements branch from c76317e to 37fb5c4CompareSeptember 12, 2020 02:24
@masahi
masahi merged commit 6a52fd4 into apache:masterSep 12, 2020
@masahi

Copy link
Copy Markdown
MemberAuthor

Thanks @siju-samuel@mbrookhart

kevinthesun pushed a commit to kevinthesun/tvm that referenced this pull request Sep 17, 2020
* support onnx GatherElements
* remove print
* run black
Co-authored-by: masa <masa@pop-os.localdomain>
kevinthesun pushed a commit to kevinthesun/tvm that referenced this pull request Sep 18, 2020
* support onnx GatherElements
* remove print
* run black
Co-authored-by: masa <masa@pop-os.localdomain>
trevor-m pushed a commit to neo-ai/tvm that referenced this pull request Sep 18, 2020
* support onnx GatherElements
* remove print
* run black
Co-authored-by: masa <masa@pop-os.localdomain>
Sign up for freeto 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

@masahi@mbrookhart@tqchen@siju-samuel