20 lines
628 B
Python
20 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
|