neural_compressor.data.datasets

Built-in datasets class for multiple framework backends.

Submodules

• neural_compressor.data.datasets.bert_dataset
• neural_compressor.data.datasets.coco_dataset
• neural_compressor.data.datasets.dataset
• neural_compressor.data.datasets.dummy_dataset
• neural_compressor.data.datasets.dummy_dataset_v2
• neural_compressor.data.datasets.imagenet_dataset
• neural_compressor.data.datasets.style_transfer_dataset

Package Contents

Classes

COCORecordDataset	Tensorflow COCO dataset in tf record format.
Datasets	A base class for all framework datasets.
Dataset	The base class of dataset.
IterableDataset	An iterable Dataset.
TensorflowImageRecord	Tensorflow imageNet database in tf record format.

Functions

dataset_registry(dataset_type, framework[, dataset_format])

Register dataset subclasses.