[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