neural_compressor.adaptor.torch_utils.onnx

Onnx Quantization Torch Utils.

Module Contents

Classes

DataReader

DataReader Class for onnx static quantization.

class neural_compressor.adaptor.torch_utils.onnx.DataReader(dataloader, sample_size=100)

Bases: ortq

DataReader Class for onnx static quantization.

get_next()

Get the next data.