pytorch

589e8a1d - [ao] Get feature and module names from ModelReportVisualizer class (#81647)

[ao] Get feature and module names from ModelReportVisualizer class (#81647) Summary: Added the functionality to be able to get the feature names and module_fqns from the ModelReportVisualizer class. The purpose of this addition is so that users can see the exact set of module_fqns or feature names that they can filter based on, and use this information to perform their filtering. Test Plan: python test/test_quantization.py TestFxModelReportVisualizer.test_get_modules_and_features Reviewers: Subscribers: Tasks: Tags: Pull Request resolved: https://github.com/pytorch/pytorch/pull/81647 Approved by: https://github.com/andrewor14