pytorch

d4f8a6d0 - [ao] Implemented report generation for InputWeightEqualization Detector (#80191)

[ao] Implemented report generation for InputWeightEqualization Detector (#80191) 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 includes the implementation and subsequent tests for the report generation functionality of the detector. The full detector should now be fleshed out and complete with this addition. This included modifications to the ModelReportObserver class as well to capture min and max per channel values. In addition, instead of passing in the observer class to instantiate, the detectors now pass the ModelReport instance the observer instance that they themselves instantiate. Test Plan: python test/test_quantization.py TestFxDetectInputWeightEqualization Reviewers: Subscribers: Tasks: Tags: Pull Request resolved: https://github.com/pytorch/pytorch/pull/80191 Approved by: https://github.com/HDCharles, https://github.com/andrewor14