neural_compressor.adaptor.ox_utils.calibration

Calibration for onnx models.

Classes

ONNXRTAugment

Augment input model to dump tensor or for calibration.

Module Contents

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.