pytorch
c5269ad6 - [quant][pt2e] Add support for a few ops in QNNPackQuantizer to enable quantizing internal model (#98560)

Commit
1 year ago
[quant][pt2e] Add support for a few ops in QNNPackQuantizer to enable quantizing internal model (#98560) Summary: This PR adds support for adaptive_avg_pool2d (traced as mean.dim), mean and hardtanh to QNNPackQuantizer Test Plan: python test/test_quantization.py TestQuantizePT2E.test_qnnpack_quantizer_obs_sharing_ops Reviewers: Subscribers: Tasks: Tags: Pull Request resolved: https://github.com/pytorch/pytorch/pull/98560 Approved by: https://github.com/andrewor14
Author
Committer
Parents
Loading