`neural_compressor.model.model`

Model for multiple framework backends.

Module Contents

Classes

A wrapper to construct a Neural Compressor Model.

Functions

get_model_fwk_name(model)

Detect the input model belongs to which framework.

neural_compressor.model.model.get_model_fwk_name(model)[source]

Detect the input model belongs to which framework.

Parameters:: model (string) – framework name that supported by Neural Compressor, if there’s no available fwk info, then return ‘NA’.

class neural_compressor.model.model.Model[source]: A wrapper to construct a Neural Compressor Model.