Skip to content

Add conv1d support in BYOC TRT by converting conv1d to conv2d#9324

Merged
vinx13 merged 1 commit into
apache:mainfrom
ziyu-guo:add-conv1d-byoc-trt
Oct 20, 2021
Merged

Add conv1d support in BYOC TRT by converting conv1d to conv2d#9324
vinx13 merged 1 commit into
apache:mainfrom
ziyu-guo:add-conv1d-byoc-trt

Conversation

@ziyu-guo

Copy link
Copy Markdown
Contributor

Currently BYOC TRT does not support conv1d due to no native conv1d in TRT. In this PR we added support for conv1d by converting it to conv2d, similar to torchtrt here.

@comaniac comaniac 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

@ziyu-guo ziyu-guo force-pushed the add-conv1d-byoc-trt branch 4 times, most recently from 44742a9 to 3ee7d34 Compare October 19, 2021 19:10
@ziyu-guo ziyu-guo force-pushed the add-conv1d-byoc-trt branch from 3ee7d34 to 515f729 Compare October 20, 2021 00:54
@vinx13 vinx13 merged commit 3f064b6 into apache:main Oct 20, 2021
ylc pushed a commit to ylc/tvm that referenced this pull request Jan 7, 2022
ylc pushed a commit to ylc/tvm that referenced this pull request Jan 13, 2022
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.

3 participants