neural_compressor.experimental.data.dataloaders.onnxrt_dataloader

Built-in dataloaders for onnxruntime framework backends.

Module Contents

Classes

ONNXRTBertDataLoader	Built-in dataloader for onnx bert model and its variants.
ONNXRTDataLoader	Built-in dataloader for onnxruntime framework backends.

class neural_compressor.experimental.data.dataloaders.onnxrt_dataloader.ONNXRTBertDataLoader

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

class neural_compressor.experimental.data.dataloaders.onnxrt_dataloader.ONNXRTDataLoader(dataset, batch_size=1, last_batch='rollover', collate_fn=None, sampler=None, batch_sampler=None, num_workers=0, pin_memory=False, shuffle=False, distributed=False)

Built-in dataloader for onnxruntime framework backends.