__init__.py 628 B

12345678910111213141516171819
  1. from .calibrate import ( # noqa: F401
  2. CalibraterBase,
  3. CalibrationDataReader,
  4. CalibrationMethod,
  5. MinMaxCalibrater,
  6. create_calibrator,
  7. )
  8. from .qdq_quantizer import QDQQuantizer # noqa: F401
  9. from .quant_utils import QuantFormat, QuantType, write_calibration_table # noqa: F401
  10. from .quantize import (
  11. DynamicQuantConfig, # noqa: F401
  12. QuantizationMode, # noqa: F401
  13. StaticQuantConfig, # noqa: F401
  14. get_qdq_config, # noqa: F401
  15. quantize, # noqa: F401
  16. quantize_dynamic, # noqa: F401
  17. quantize_static, # noqa: F401
  18. )
  19. from .shape_inference import quant_pre_process # noqa: F401