`neural_compressor.experimental.mixed_precision`¶

Class for low precision model generation across multiple framework backends.

Module Contents¶

Class used for generating low precision model.

class neural_compressor.experimental.mixed_precision.MixedPrecision(conf_fname_or_obj=None)¶

Bases: neural_compressor.experimental.graph_optimization.GraphOptimization

Class used for generating low precision model.

MixedPrecision class automatically generates low precision model across various DL frameworks including tensorflow, pytorch and onnxruntime.

property precisions¶: Get private member variable precisions of MixedPrecision class.

set_config_by_model(model_obj)¶: Set member variable conf by a input model object.