neural_compressor.model.model

Model for multiple framework backends.

Classes

Model

A wrapper to construct a Neural Compressor Model.

Functions

get_model_fwk_name(model)

Detect the input model belongs to which framework.

Module Contents

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.