pytorch
4f94c8e0 - [ao] Implemented InputWeightEqualization Detector observer insertion (#79962)

Commit
3 years ago
[ao] Implemented InputWeightEqualization Detector observer insertion (#79962) Summary: This adds the implementation for the InputWeightEqualization detector. This includes both the implementation and the relavent test cases. This detector is meant to be added to initialize a ModelReport instance and it will keep track of the necessary statistics to decide if for certain layers of interest (linear and conv for now), it makes sense to use input weight equalization and gives the suggestion to the user. This implements the functionality of adding observer points for the input-weight equalization detector and contains the relavent tests for this functionality. The full Detector functionality will be fleshed out in a later commit. Test Plan: python test/test_quantization.py TestFxDetectInputWeightEqualization Reviewers: Subscribers: Tasks: Tags: Pull Request resolved: https://github.com/pytorch/pytorch/pull/79962 Approved by: https://github.com/HDCharles, https://github.com/andrewor14
Author
Committer
Parents
Loading