Skip to content

[Relay][Frontend][Onnx] Enable group_conv1d import through conv2d conversion. - #8321

Merged
mbrookhart merged 2 commits into
apache:mainfrom
jwfromm:group_conv1d_onnx
Jun 24, 2021
Merged

[Relay][Frontend][Onnx] Enable group_conv1d import through conv2d conversion.#8321
mbrookhart merged 2 commits into
apache:mainfrom
jwfromm:group_conv1d_onnx

Conversation

@jwfromm

@jwfromm jwfromm commented Jun 24, 2021

Copy link
Copy Markdown
Contributor

TVM doesn't support group_conv1d currently and adding it formally is a non-trivial amount of work. Meanwhile, we're encountering an increasing number of models that use it. As an interim solution, we can import group_conv1d using group_conv2d by simply adding a new spatial axis with size 1. This PR adds that converter into the onnx frontend along with corresponding tests.

@jwfromm
jwfromm requested a review from mbrookhart June 24, 2021 00:55
@jwfromm

jwfromm commented Jun 24, 2021

Copy link
Copy Markdown
Contributor Author

@mbrookhart @masahi can you guys take a look at this PR?

@mbrookhart mbrookhart 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.

Nit, but needs to be changed. Otherwise looks good. I appreciate the TODO.

In general, and for what it's worth, I find the -k option in pytest easier than commenting out the test list 🤷‍♂️

Comment thread tests/python/frontend/onnx/test_forward.py Outdated
@mbrookhart
mbrookhart merged commit 6b7b966 into apache:main Jun 24, 2021
ylc pushed a commit to ylc/tvm that referenced this pull request Sep 29, 2021
…version. (apache#8321)

* Enable group conv1d import through conv2d hack.

* remove silly commented out lines.
zxy844288792 pushed a commit to zxy844288792/tvm that referenced this pull request Mar 4, 2022
…version. (apache#8321)

* Enable group conv1d import through conv2d hack.

* remove silly commented out lines.
@jwfromm
jwfromm deleted the group_conv1d_onnx branch April 12, 2023 15:55
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