pytorch
471407cf - [PT2][Quant] Use composble quantizer for embedding + static conv + dynamic (#103116)

Commit
1 year ago
[PT2][Quant] Use composble quantizer for embedding + static conv + dynamic (#103116) Summary: In this diff we test a module that does a) emedding lookup b) runs 1D (converted to 2D) conv and c) runs linear on the output of 1d conv. a is quantized using embedding quantizer. c is quantized using dynamic quantization. b is quantized using static quantization. We compose quantizer from [a, c, b]. Tested it against similar fx config. Test Plan: test_embedding_conv_linear_quantization Reviewed By: jerryzh168 Differential Revision: D46267688 Pull Request resolved: https://github.com/pytorch/pytorch/pull/103116 Approved by: https://github.com/jerryzh168
Author
Committer
Parents
Loading