pytorch
e69115ec - [quant][graph] Add JIT passes for dynamic quant multi uses of quant node (#37125)

Commit
5 years ago
[quant][graph] Add JIT passes for dynamic quant multi uses of quant node (#37125) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/37125 For dynamic quant we need to replicate the choose_qparams and quantize function in addition to replicating dequant. RemoveRedundantQuantizeOps pass checks for the choose_qparams - quant - dequant pattern in the graph and removes it if the node following it cannot be quantized using dynamic quantization. Test Plan: python test_quantize_script.py test_dynamic_quant_multi_uses Imported from OSS Differential Revision: D21283697 fbshipit-source-id: 70fa0abdaeb2cc2935149a941d93a7e8b28d61d3
Author
Parents
Loading