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

class neural_compressor.model.Model

Bases: object

A wrapper to construct a Neural Compressor Model.

class neural_compressor.model.BaseModel(model, **kwargs)

Base class of all neural_compressor.model, will play graph role.

abstract property model

Return model itself.

abstract property graph_info

Node_type’.

Type:

Return a dict with content ‘Node

abstract save(root, *args, **kwargs)

Abstract method of model saving.

abstract export(save_path: str, conf)

Abstract method of model convertion to ONNX.

abstract framework()

Abstract method of model framework.