neural_compressor.experimental.data.datasets

Built-in datasets class for multiple framework backends.

Submodules

• 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

Package Contents

Classes

Datasets	A base class for all framework datasets.
Dataset	The base class of dataset.
IterableDataset	An iterable Dataset.

Functions

dataset_registry(dataset_type, framework[, dataset_format])

Register dataset subclasses.