Searched refs:ModelReportVisualizer (Results 1 – 4 of 4) sorted by relevance
24 class ModelReportVisualizer: class492 len(tensor_headers) - ModelReportVisualizer.NUM_NON_FEATURE_TENSOR_HEADERS495 len(channel_headers) - ModelReportVisualizer.NUM_NON_FEATURE_CHANNEL_HEADERS503 feature_column_offset = ModelReportVisualizer.NUM_NON_FEATURE_TENSOR_HEADERS509 ModelReportVisualizer.NUM_NON_FEATURE_CHANNEL_HEADERS
16 ModelReportVisualizer,432 def generate_visualizer(self) -> ModelReportVisualizer:452 visualizer: ModelReportVisualizer = ModelReportVisualizer(
41 # Optional: we get a ModelReportVisualizer instance to do any visualizations desired67 ## ModelReportVisualizer Overview70 you can visualize some of the collected statistics using the `ModelReportVisualizer`.71 To get a `ModelReportVisualizer` instance from the `ModelReport` instance,74 When you first create the `ModelReportVisualizer` instance,181 - `model_report_visualizer.py`: File containing the `ModelReportVisualizer` class200 - Test class for the `ModelReportVisualizer` class
16 from torch.ao.quantization.fx._model_report.model_report_visualizer import ModelReportVisualizer1130 mod_rep_visualizer: ModelReportVisualizer = mod_report.generate_visualizer()1776 mod_rep_visualizer: ModelReportVisualizer = mod_report.generate_visualizer()1803 mod_rep_visualizer: ModelReportVisualizer = self._callibrate_and_generate_visualizer(1850 mod_rep_visualizer: ModelReportVisualizer = self._callibrate_and_generate_visualizer(1873 tensor_headers, tensor_table = table_dict[ModelReportVisualizer.TABLE_TENSOR_KEY]1874 channel_headers, channel_table = table_dict[ModelReportVisualizer.TABLE_CHANNEL_KEY]1901 tensor_headers, tensor_table = empty_tables_dict[ModelReportVisualizer.TABLE_TENSOR_KEY]1902 … channel_headers, channel_table = empty_tables_dict[ModelReportVisualizer.TABLE_CHANNEL_KEY]1927 tensor_headers, tensor_table = single_feat_dict[ModelReportVisualizer.TABLE_TENSOR_KEY][all …]