onnxruntime/onnxruntime/python/tools/quantization/__init__.py
Justin Chu c7c8757a1c
Use ruff as the formatter to replace black-isort (#23397)
Use ruff as the code formatter in place of black and isort since it is
much faster, and as projects like PyTorch and ONNX have adopted ruff
format as well.

This PR include only auto-fixed changes in formatting.
2025-01-16 11:14:15 -08:00

19 lines
628 B
Python

from .calibrate import ( # noqa: F401
CalibraterBase,
CalibrationDataReader,
CalibrationMethod,
MinMaxCalibrater,
create_calibrator,
)
from .qdq_quantizer import QDQQuantizer # noqa: F401
from .quant_utils import QuantFormat, QuantType, write_calibration_table # noqa: F401
from .quantize import (
DynamicQuantConfig, # noqa: F401
QuantizationMode, # noqa: F401
StaticQuantConfig, # noqa: F401
get_qdq_config, # noqa: F401
quantize, # noqa: F401
quantize_dynamic, # noqa: F401
quantize_static, # noqa: F401
)
from .shape_inference import quant_pre_process # noqa: F401