up

Author: metascroy

backends/xnnpack/quantizer/xnnpack_quantizer_utils.py

@@ -23,6 +23,7 @@
     OperatorPatternType,
     QuantizationAnnotation,
     QuantizationConfig,
+    QuantizationSpec,
     SharedQuantizationSpec,
 )
 from torchao.quantization.pt2e.utils import (
@@ -36,6 +37,7 @@
     "OperatorConfig",
     "OperatorPatternType",
     "QuantizationConfig",
+    "QuantizationSpec",
     "get_input_act_qspec",
     "get_output_act_qspec",
     "get_weight_qspec",