neural_compressor.model

Built-in model for multiple framework backends.

Submodules

• neural_compressor.model.base_model
• neural_compressor.model.keras_model
• neural_compressor.model.model
• neural_compressor.model.mxnet_model
• neural_compressor.model.nets_factory
• neural_compressor.model.onnx_model
• neural_compressor.model.tensorflow_model
• neural_compressor.model.torch_model

Package Contents

Classes

Model	A wrapper to construct a Neural Compressor Model.
BaseModel	Base class of all neural_compressor.model, will play graph role.

Attributes

MODELS