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

[CMSIS-NN] Convert scalar constants to tensor constants #10100

Merged
merged 4 commits into from
Feb 2, 2022

Conversation

ashutosh-arm
Copy link
Contributor

  • Scalar to tensor constant pass

-Pass that converts scalar constants into tensor constants for CMSIS-NN binary ops
-Updated extract_constants pass to support constants at any position in the function signature
-Modified tests for extract_constants pass corresponding to the scalar support
-Added new binary op tests to include constants

Change-Id: I9ea9c28b1410b4a80a9235af2e84bc80b4dc3a66
Change-Id: Ib916e4b29be124ccdf4c88936f10499b6642f141
Copy link
Member

@Mousius Mousius left a comment

Choose a reason for hiding this comment

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

First pass over this @ashutosh-arm, thanks for fixing this 😸

src/relay/backend/contrib/cmsisnn/extract_constants.cc Outdated Show resolved Hide resolved
src/relay/backend/contrib/cmsisnn/extract_constants.cc Outdated Show resolved Hide resolved
src/relay/backend/contrib/cmsisnn/extract_constants.cc Outdated Show resolved Hide resolved
src/relay/backend/contrib/cmsisnn/extract_constants.cc Outdated Show resolved Hide resolved
tests/python/contrib/test_cmsisnn/test_binary_ops.py Outdated Show resolved Hide resolved
…hon functions, expanded variable names

Change-Id: Ieacbe7b523e7ca4db54f15b4b7da117f421ec36b
@ashutosh-arm
Copy link
Contributor Author

Thanks for the review @Mousius. I have updated the PR with review comments.

@Mousius Mousius self-assigned this Feb 1, 2022
Change-Id: I625d5db043c987c46ca3521c3ce94873694d4d87
Copy link
Member

@Mousius Mousius left a comment

Choose a reason for hiding this comment

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

LGTM!

@Mousius Mousius merged commit cb3d7e2 into apache:main Feb 2, 2022
@Mousius
Copy link
Member

Mousius commented Feb 2, 2022

Thanks @ashutosh-arm! This is merged now 😸

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