Bam17
/ toy_example_quantization_2_0.py
Last active
March 20, 2024 18:30
— forked from leslie-fang-intel/toy_example_quantization_2_0.py
Toy Example code for Quantization in PyTorch 2.0 Export Tutorial
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import copy | |
| import itertools | |
| import operator | |
| from typing import Callable, Dict, List, Optional, Set, Any | |
| import torch | |
| import torch._dynamo as torchdynamo | |
| from torch.ao.quantization.quantizer.utils import ( | |
| _annotate_input_qspec_map, | |
| _annotate_output_qspec) |