Home
last modified time | relevance | path

Searched refs:ModelReportVisualizer (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/ao/quantization/fx/_model_report/
H A Dmodel_report_visualizer.py24 class ModelReportVisualizer: class
492 len(tensor_headers) - ModelReportVisualizer.NUM_NON_FEATURE_TENSOR_HEADERS
495 len(channel_headers) - ModelReportVisualizer.NUM_NON_FEATURE_CHANNEL_HEADERS
503 feature_column_offset = ModelReportVisualizer.NUM_NON_FEATURE_TENSOR_HEADERS
509 ModelReportVisualizer.NUM_NON_FEATURE_CHANNEL_HEADERS
H A Dmodel_report.py16 ModelReportVisualizer,
432 def generate_visualizer(self) -> ModelReportVisualizer:
452 visualizer: ModelReportVisualizer = ModelReportVisualizer(
H A DREADME.md41 # Optional: we get a ModelReportVisualizer instance to do any visualizations desired
67 ## ModelReportVisualizer Overview
70 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` class
200 - Test class for the `ModelReportVisualizer` class
/aosp_15_r20/external/pytorch/test/quantization/fx/
H A Dtest_model_report_fx.py16 from torch.ao.quantization.fx._model_report.model_report_visualizer import ModelReportVisualizer
1130 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 …]