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

[VM] Fix the shape function of conv & Add dynamic support for conv2d nhwc #8480

Merged
merged 8 commits into from
Jul 16, 2021

Conversation

jcf94
Copy link
Contributor

@jcf94 jcf94 commented Jul 15, 2021

Just a small bug fix when using vm with conv2d_nhwc. Also add the shape function for nn.fast_softmax.

cc @comaniac

Copy link
Contributor

@comaniac comaniac left a comment

Choose a reason for hiding this comment

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

LGTM. Just some style suggestions.

python/tvm/relay/op/nn/_nn.py Outdated Show resolved Hide resolved
@jcf94 jcf94 merged commit cba9cf3 into apache:main Jul 16, 2021
@jcf94 jcf94 deleted the conv2d_nhwc branch July 16, 2021 12:22
ylc pushed a commit to ylc/tvm that referenced this pull request Sep 29, 2021
* Add dynamic support for conv2d nhwc
zxy844288792 pushed a commit to zxy844288792/tvm that referenced this pull request Mar 4, 2022
* Add dynamic support for conv2d nhwc
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