core.quantization.utils

Module Contents

Functions

get_quant_config_or_none	Resolve quantization config for a layer.
load_quantization_recipe	Loads a quantization recipe from a path.
kitchen_quantization_recipe_config	Loads a quantization recipe that uses a QAT_PARAMS recipe for all layers.

API

core.quantization.utils.get_quant_config_or_none(

module_path: str,

recipe: Optional[core.quantization.quant_config.RecipeConfig] = None,

) → Union[core.quantization.quant_config.QuantizationConfig, None]

Resolve quantization config for a layer.

core.quantization.utils.load_quantization_recipe(

recipe_path: str,

) → core.quantization.quant_config.RecipeConfig

Loads a quantization recipe from a path.

core.quantization.utils.kitchen_quantization_recipe_config(

recipe_idx: int,

) → core.quantization.quant_config.RecipeConfig

Loads a quantization recipe that uses a QAT_PARAMS recipe for all layers.