neural_compressor.data.dataloaders.onnxrt_dataloader

Built-in dataloaders for onnxruntime framework backends.

Classes

ONNXRTBertDataLoader

Built-in dataloader for onnx bert model and its variants.

ONNXRTDataLoader

Built-in dataloader for onnxruntime framework backends.

Module Contents

class neural_compressor.data.dataloaders.onnxrt_dataloader.ONNXRTBertDataLoader[source]

Built-in dataloader for onnx bert model and its variants.

class neural_compressor.data.dataloaders.onnxrt_dataloader.ONNXRTDataLoader[source]

Built-in dataloader for onnxruntime framework backends.