pytorch
74291c83 - [quant][graphmode][fx] Fix the calls to load_arg in quantization_patterns.py (#62376)

Commit
3 years ago
[quant][graphmode][fx] Fix the calls to load_arg in quantization_patterns.py (#62376) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/62376 load_arg(quantized=...) accepts a dictionary from index to dtype, not a list of dtype, the call is just to make sure the inputs are quantized with correct dtype Test Plan: python test/test_quantization.py TestQuantizeFx python test/test_quantization.py TestQuantizeFxOps Imported from OSS Reviewed By: heitorschueroff Differential Revision: D29979711 fbshipit-source-id: 8499976ac5df8eb2019c3beae573dec6c9a56247
Author
Parents
Loading