pytorch/test/quantization
Kimish Patel cfd18e105f [Pytorch][Ondevice quantization] Add device side API to convert model (#83807)
Summary:
This diff adds device side API which will convert the model to its
quantized equivalent. THe input model must have been prepared AOT for
quantization.

API is implemented by:
- Running reset obervers
- Running observe method
- Running quantize method
- And replacing method, e.g. forward, with its quantized equivalent.

Test Plan:
test/quantization/jit/test_ondevice_quantization.py

Reviewers:

Subscribers:

Tasks:

Tags:

Differential Revision: [D38889818](https://our.internmc.facebook.com/intern/diff/D38889818)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/83807
Approved by: https://github.com/iseeyuan
2022-08-29 17:57:38 +00:00
..
ao_migration [quant][ao_migration] torch.nn.qattorch.ao.nn.qat (#78716) 2022-08-25 16:50:38 +00:00
bc [quant][ao_migration] torch.nn.quantized.dynamictorch.ao.nn.quantized.dynamic (#78714) 2022-08-25 16:50:34 +00:00
core [quant][ao_migration] torch.nn.quantized._referencetorch.ao.nn.quantized._reference (#78715) 2022-08-25 16:50:36 +00:00
eager conv2d: require bias to have the same dtype as input and weight on cpu (#83686) 2022-08-29 16:41:17 +00:00
fx [quant][ao_migration] torch.nn.qattorch.ao.nn.qat (#78716) 2022-08-25 16:50:38 +00:00
jit [Pytorch][Ondevice quantization] Add device side API to convert model (#83807) 2022-08-29 17:57:38 +00:00
serialized
__init__.py