File tree Expand file tree Collapse file tree 2 files changed +4
-0
lines changed
nncf/experimental/torch/fx/quantization Expand file tree Collapse file tree 2 files changed +4
-0
lines changed Original file line number Diff line number Diff line change 2525import nncf
2626from nncf .common .factory import NNCFGraphFactory
2727from nncf .common .logging import nncf_logger
28+ from nncf .common .utils .api_marker import api
2829from nncf .data import Dataset
2930from nncf .experimental .quantization .algorithms .post_training .algorithm import ExperimentalPostTrainingQuantization
3031from nncf .experimental .torch .fx .constant_folding import constant_fold
3839from nncf .quantization .advanced_parameters import RangeEstimatorParameters
3940
4041
42+ @api ()
4143def quantize_pt2e (
4244 model : torch .fx .GraphModule ,
4345 quantizer : Quantizer ,
Original file line number Diff line number Diff line change 3030from nncf .common .quantization .quantizer_setup import SingleConfigQuantizerSetup
3131from nncf .common .quantization .structs import QuantizationPreset
3232from nncf .common .quantization .structs import QuantizationScheme
33+ from nncf .common .utils .api_marker import api
3334from nncf .experimental .torch .fx .nncf_graph_builder import GraphConverter
3435from nncf .experimental .torch .fx .node_utils import get_graph_node_by_name
3536from nncf .experimental .torch .fx .transformations import fold_constant_except_qdq
4647QUANT_ANNOTATION_KEY = "quantization_annotation"
4748
4849
50+ @api ()
4951class OpenVINOQuantizer (TorchAOQuantizer ):
5052 """
5153 Implementation of the Torch AO quantizer which annotates models with quantization annotations
You can’t perform that action at this time.
0 commit comments