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] Support ConvTranspose + auto_pad #3608

Closed
stakemura opened this issue Jul 24, 2019 · 7 comments
Closed

[ONNX] Support ConvTranspose + auto_pad #3608

stakemura opened this issue Jul 24, 2019 · 7 comments

Comments

@stakemura
Copy link

stakemura commented Jul 24, 2019

TVM Relay (latest master branch) couldn't compile ONNX including ConvTranspose + auto_pad.

Here is the code and the attached
ZIP-compressed ONNX file updated on Jul 25
to reproduce the error.

import tvm
import tvm.relay as relay
import onnx

onnx_model = onnx.load('upconvert.onnx')
 
shape_dict = {'input__0': (1, 3, 360, 640)}
mod, params = relay.frontend.from_onnx(onnx_model, shape_dict)

This issue is critical for us.
Would you like to support this attribute?

@tqchen tqchen changed the title [Relay] Failed to compile ONNX including ConvTranspose + auto_pad [ONNX] Support ConvTranspose + auto_pad Jul 24, 2019
@tqchen
Copy link
Member

tqchen commented Jul 24, 2019

It would be great if you can look a bit into what is going on, and the community can provide help on possibly actions

@vinx13
Copy link
Member

vinx13 commented Jul 25, 2019

abuccts added a commit to abuccts/tvm that referenced this issue Jul 28, 2019
Support `auto_pad` attribute in ONNX Conv/ConvTranspose operators,
infer pads shape for `auto_pad`.

Fixes apache#3608.
@stakemura
Copy link
Author

@vinx13 Thank you for the supplementary explanation!
@abuccts I deeply appreciate your help. I would like to try your patch.

@abuccts
Copy link
Contributor

abuccts commented Aug 7, 2019

Hi @stakemura, could you help to verify the patch? @vinx13 could you please review the PR, thx!

@stakemura
Copy link
Author

@abuccts I'm sorry for the late reply. I confirmed #3645 patch can load the attached onnx model successfully.

@cchung100m
Copy link
Contributor

Hi @zhiics

This PR should be closed by #4563, too. I would appreciate that if you can help to manage it, thanks :)

@zhiics
Copy link
Member

zhiics commented Jan 6, 2020

@cchung100m Thanks for resolving the issue.

@zhiics zhiics closed this as completed Jan 6, 2020
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging a pull request may close this issue.

6 participants