pytorch
cfd18e10 - [Pytorch][Ondevice quantization] Add device side API to convert model (#83807)

Commit
3 years ago
[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
Author
Committer
Parents
Loading