pytorch/test/quantization
Jerry Zhang 8f88f797db [quant][graphmode][fx] Add reference quantized conv module (#63828)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/63828

Added reference quantized conv module for the custom backend flow, the reference quantized module will
have the following code:
```
        w(float) -- quant - dequant \
        x(float) ------------- F.conv2d ---
```
In the full model, we will see
```
        w(float) -- quant - *dequant \
        x -- quant --- *dequant --  *F.conv2d --- *quant - dequant
```
and the backend should be able to fuse the ops with `*` into a quantized linear

Test Plan:
python test/test_quantization.py TestQuantizeFx.test_conv_linear_reference

Imported from OSS

Reviewed By: vkuzo

Differential Revision: D30504749

fbshipit-source-id: e1d8c43a0e0d6d9ea2375b8ca59a9c0f455514fb
2021-08-30 14:23:17 -07:00
..
ao_migration [quant] AO migration of the quantize.py (#64086) 2021-08-29 20:30:01 -07:00
bc [quant] Make version 1 the default for get_default_qat_qconfig (#63043) 2021-08-11 22:06:44 -07:00
core [quant][graphmode][fx] Add reference quantized conv module (#63828) 2021-08-30 14:23:17 -07:00
eager [quant][fx] Add support for dynamic linear + relu fusion (INT8) (#63799) 2021-08-26 21:10:46 -07:00
fx [quant][graphmode][fx] Add reference quantized conv module (#63828) 2021-08-30 14:23:17 -07:00
jit Updates internal assert_allclose callsites in favor of assert_close (#61841) 2021-08-19 12:50:41 -07:00
serialized [quant] Make version 1 the default for get_default_qat_qconfig (#63043) 2021-08-11 22:06:44 -07:00
__init__.py