pytorch/test/quantization/core
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
..
__init__.py
test_quantized_functional.py
test_quantized_module.py [quant][graphmode][fx] Add reference quantized conv module (#63828) 2021-08-30 14:23:17 -07:00
test_quantized_op.py [quant] Add op support for linear_relu_dynamic_fp16 (#63824) 2021-08-26 21:12:04 -07:00
test_quantized_tensor.py BFP16 quantization/dequantization (#62974) 2021-08-09 15:47:07 -07:00
test_workflow_module.py replace self.assertTrue(torch.allclose(..)) with self.assertEqual(…) (#63637) 2021-08-25 16:47:40 -07:00
test_workflow_ops.py replace self.assertTrue(torch.allclose(..)) with self.assertEqual(…) (#63637) 2021-08-25 16:47:40 -07:00