pytorch
ce9cedd1 - [quant] Input-Weight Equalization - Conv observer support (#61285)

Commit
3 years ago
[quant] Input-Weight Equalization - Conv observer support (#61285) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/61285 Modifies observers to support conv layers and tests to make sure that the observers are returning the expected values for conv inputs. Test Plan: `python test/test_quantization.py TestEqualizeFx.test_input_weight_eq_observer` Imported from OSS Reviewed By: jerryzh168 Differential Revision: D29557041 fbshipit-source-id: 5e43329f189ba352eb8b991f38bf37752eebb6e6
Author
Parents
Loading