pytorch
ddd916c2 - [quant][refactor] Return the models in checkGraphModeFxOp for further checking (#62487)

Commit
3 years ago
[quant][refactor] Return the models in checkGraphModeFxOp for further checking (#62487) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/62487 checkGraphModeFxOp is our utility test function to quantize a given model with FX Graph Mode Quantization and checks whether the result model contains expected ops, previously it only returns a result on the sample data for the quantized model, this PR chagnes it to return prepared, quantized, quantized_reference models together with the result for quantized models. Test Plan: python test/test_quantization.py TestQuantizeFx python test/test_quantization.py TestQuantizeFxOps Imported from OSS Reviewed By: iramazanli Differential Revision: D30053981 fbshipit-source-id: 31fbce48d138261d0b00ba24e1427fd0c6208990
Author
Parents
Loading