mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-21 21:52:11 +00:00
ffc2b25a3a
Improve quantization tools: 1. Support QAT 2. Make quantization tool to register Operators. 3. Make the API clear to use Co-authored-by: t-yguo <t-yguo@microsoft.com>
5 lines
232 B
Python
5 lines
232 B
Python
from .quantize import quantize, quantize_static, quantize_dynamic, quantize_qat
|
|
from .quantize import QuantizationMode
|
|
from .calibrate import CalibrationDataReader
|
|
from .calibrate import calibrate
|
|
from .quant_utils import QuantType
|