pytorch
5a8198eb - [quant][graphmode][fx][fix] scalar as first input for add/mul (#46751)

Commit
4 years ago
[quant][graphmode][fx][fix] scalar as first input for add/mul (#46751) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/46751 Currently we assume the first input for add/mul is node (Tensor), but it might not be the case Test Plan: python test/test_quantization.py TestQuantizeFxOps.test_quantized_add python test/test_quantization.py TestQuantizeFxOps.test_quantized_mul python test/test_quantization.py TestQuantizeFxOps.test_quantized_add_relu python test/test_quantization.py TestQuantizeFxOps.test_quantized_mul_relu Imported from OSS Reviewed By: raghuramank100 Differential Revision: D24494456 fbshipit-source-id: ef5e23ba60eb22a57771791f4934306b25c27c01
Author
Parents
Loading