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

[microNPU] Optimize separate padding operation for conv2d #11468

Merged
merged 1 commit into from
Jun 6, 2022

Conversation

lhutton1
Copy link
Contributor

@lhutton1 lhutton1 commented May 26, 2022

Optimizes a case where padding appears as a separate nn.pad operation followed by a qnn.conv2d. If possible, the nn.pad will be partitioned and offloaded together with the qnn.conv2d operation, as opposed to separately. As a fallback, both operations will be considered separately.

cc @Mousius @NicolaLancellotti @ekalda @manupa-arm

Optimizes a case where padding appears as a separate nn.pad operation
followed by a qnn.conv2d. If possible, the nn.pad will be partitioned
and offloaded together with the qnn.conv2d operation, as opposed to
separately. As a fallback, both operations will be considered
separately.

Change-Id: I9125195386abdcc1d17ec612dc0a4cd6474d637a
@github-actions github-actions bot requested a review from manupak May 26, 2022 13:18
Copy link
Contributor

@manupak manupak left a comment

Choose a reason for hiding this comment

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

LGTM!

@lhutton1
Copy link
Contributor Author

lhutton1 commented Jun 6, 2022

@tvm-bot merge

@github-actions github-actions bot merged commit 1aac4d6 into apache:main Jun 6, 2022
@lhutton1 lhutton1 deleted the optimize-pad-conv2d branch June 6, 2022 14:10
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