pytorch

04fc3e4c - [ao] Add histogram visualization capability to ModelReportVisualizer (#81975)

[ao] Add histogram visualization capability to ModelReportVisualizer (#81975) Summary: This adds the capability to visualize the histogram in the ModelReportVisualizer. You can visualize the histogram of a single feature for a single layer, (for example you want to see the distribution of some data across all channels), or for some feature across multiple layers of a similar kind. All channel data is merged together to plot one large distribution. The user gets to decide the number of bins the histogram has and it will create those many equally spaced bins. Expected Usage ``` mod_rep_visualizer.generate_histogram_visualization(<feature_name>,<module_name>) ``` You can also filter the modules so that only modules with a certain substring will have their features represented in the plot. > ** This is intended to be used in a `.ipynb` style notebook** The tests for this were just visual inspection for two reasons: 1.) This method does not return anything, it just generates the visualization plot 2.) All the data to create the plot visualization is gotten from `generate_filtered_tables` which is already tested, so testing all that for this again would be redundant. Example Image outputs are pasted below in the PR thread. Test Plan: Visual Test Reviewers: Subscribers: Tasks: Tags: Pull Request resolved: https://github.com/pytorch/pytorch/pull/81975 Approved by: https://github.com/jerryzh168