`neural_compressor.adaptor.ox_utils.calibration`

Calibration for onnx models.

Module Contents

Classes

ONNXRTAugment

Augment input model to dump tensor or for calibration.

class neural_compressor.adaptor.ox_utils.calibration.ONNXRTAugment(model_wrapper, dataloader, dump_op_types, black_nodes=[], white_nodes=[], iterations=[], backend='CPUExecutionProvider', reduce_range=False, **kwargs)[source]: Augment input model to dump tensor or for calibration.