Skip to content

[QNN] Support input scale and zp of 1-element vector in qnn.conv2d_transpose#10952

Merged
AndrewZhaoLuo merged 2 commits intoapache:mainfrom
sfvaroglu:sevin/qnn_conv2d_transpose_shape
Apr 20, 2022
Merged

[QNN] Support input scale and zp of 1-element vector in qnn.conv2d_transpose#10952
AndrewZhaoLuo merged 2 commits intoapache:mainfrom
sfvaroglu:sevin/qnn_conv2d_transpose_shape

Conversation

@sfvaroglu
Copy link
Contributor

We are seeing some models with 1-element vectors for input_scale and input_zero_point in qnn.conv2d_transpose. This PR adds support to handle those shapes.

@sfvaroglu
Copy link
Contributor Author

@AndrewZhaoLuo @anwang2009 please review

Copy link
Contributor

@AndrewZhaoLuo AndrewZhaoLuo left a comment

Choose a reason for hiding this comment

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

Hmm curious your thoughts. One more comment, please rebase too.

@sfvaroglu sfvaroglu force-pushed the sevin/qnn_conv2d_transpose_shape branch from 759555d to 3c12438 Compare April 18, 2022 23:10
Copy link
Contributor

@AndrewZhaoLuo AndrewZhaoLuo left a comment

Choose a reason for hiding this comment

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

LGTM, need to read more into broadcasting conventions in TVM

@AndrewZhaoLuo AndrewZhaoLuo merged commit 58b7a5a into apache:main Apr 20, 2022
shtinsa pushed a commit to Deelvin/tvm that referenced this pull request May 17, 2022
…anspose (apache#10952)

* Support input scale and zp of 1-element vector in qnn.conv2d_transpose

* Lint
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