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

[Relay] Add support of conv2d with NHWC for Mali #8422

Merged
merged 2 commits into from
Jul 9, 2021

Conversation

AnastasiaStulova
Copy link
Contributor

Added template schedule for conv2d NHWC reusing similar strategy
as for NCHW layout. The schedule is also added to the
corresponding test that can be run to verify correctness.

Added template schedule for conv2d NHWC reusing similar strategy
as for NCHW layout. The schedule is also added to the
corresponding test that can be run to verify correctness.
@AnastasiaStulova
Copy link
Contributor Author

NOTE that this patch need the fix in #8361 to pass the test cases with deletion > 1.

@AnastasiaStulova
Copy link
Contributor Author

@mbaret

@AnastasiaStulova
Copy link
Contributor Author

@merrymercy @i4oolish

Copy link
Contributor

@jcf94 jcf94 left a comment

Choose a reason for hiding this comment

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

LGTM. Thanks! @AnastasiaStulova
Let's merge #8361 first.

@jcf94 jcf94 merged commit e7c5349 into apache:main Jul 9, 2021
ylc pushed a commit to ylc/tvm that referenced this pull request Sep 29, 2021
* [Relay] Add support of conv2d with NHWC for Mali

Added template schedule for conv2d NHWC reusing similar strategy
as for NCHW layout. The schedule is also added to the
corresponding test that can be run to verify correctness.

* [Relay] Fix issue from pylint in conv2d for Mali
zxy844288792 pushed a commit to zxy844288792/tvm that referenced this pull request Mar 4, 2022
* [Relay] Add support of conv2d with NHWC for Mali

Added template schedule for conv2d NHWC reusing similar strategy
as for NCHW layout. The schedule is also added to the
corresponding test that can be run to verify correctness.

* [Relay] Fix issue from pylint in conv2d for Mali
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