`neural_compressor.onnxrt.quantization.calibrate`

Module Contents

Classes

CalibrationDataReader

Get data for calibration.

class neural_compressor.onnxrt.quantization.calibrate.CalibrationDataReader[source]

Get data for calibration.

We define our CalibrationDataReader based on the class in below link: https://github.com/microsoft/onnxruntime/blob/main/onnxruntime/python/tools/quantization/calibrate.py#L139