neural_compressor.tensorflow.quantization.algorithm_entry

Module Contents

Functions

static_quant_entry(model, quant_config[, ...])

The main entry to apply static quantization.
neural_compressor.tensorflow.quantization.algorithm_entry.static_quant_entry(model: neural_compressor.tensorflow.utils.BaseModel, quant_config: neural_compressor.common.base_config.BaseConfig, calib_dataloader: Callable = None, calib_iteration: int = 100)[source]

The main entry to apply static quantization.

Parameters:

  • model – a fp32 model to be quantized.

  • quant_config – a quantization configuration.

  • calib_dataloader – a data loader for calibration.

  • calib_iteration – the iteration of calibration.

Returns:

the quantized model.

Return type:

q_model