neural_compressor.experimental.data
Built-in dataloaders, datasets, transforms, filters for multiple framework backends.
Subpackages
neural_compressor.experimental.data.dataloaders
neural_compressor.experimental.data.dataloaders.base_dataloader
neural_compressor.experimental.data.dataloaders.dataloader
neural_compressor.experimental.data.dataloaders.default_dataloader
neural_compressor.experimental.data.dataloaders.fetcher
neural_compressor.experimental.data.dataloaders.mxnet_dataloader
neural_compressor.experimental.data.dataloaders.onnxrt_dataloader
neural_compressor.experimental.data.dataloaders.pytorch_dataloader
neural_compressor.experimental.data.dataloaders.sampler
neural_compressor.experimental.data.dataloaders.tensorflow_dataloader
neural_compressor.experimental.data.datasets
neural_compressor.experimental.data.datasets.bert_dataset
neural_compressor.experimental.data.datasets.coco_dataset
neural_compressor.experimental.data.datasets.dataset
neural_compressor.experimental.data.datasets.dummy_dataset
neural_compressor.experimental.data.datasets.dummy_dataset_v2
neural_compressor.experimental.data.datasets.imagenet_dataset
neural_compressor.experimental.data.datasets.style_transfer_dataset
neural_compressor.experimental.data.filters
neural_compressor.experimental.data.transforms
Package Contents
Classes
|
A base class for all framework datasets. |
|
The base class of dataset. |
|
An iterable Dataset. |
|
Transforms collection class. |
|
The base class for transform. |
|
The filter register for all frameworks. |
|
The base class for transform. |
Functions
|
Register dataset subclasses. |
|
Class decorator used to register all transform subclasses. |
|
Register all transform subclasses. |
Attributes
|