coreai_opt.quantization.spec.GlobalMinMaxQParamsCalculator¶
- class coreai_opt.quantization.spec.GlobalMinMaxQParamsCalculator(*args, **kwargs)[source]¶
Bases:
RunningRangeMixin,QParamsCalculatorBaseComputes scale and zero point by tracking the running min/max.
Maintains
running_minandrunning_maxbuffers that are updated each forward pass via element-wise minimum and maximum:running_min = min(running_min, x_min) running_max = max(running_max, x_max)
- Parameters:
args (object)
kwargs (object)
- __init__(*args, **kwargs)¶
- Parameters:
args (object)
kwargs (object)
- Return type:
None
Methods
compute_qparams(tensor, min_val, max_val)Update running range, persist to buffers, then compute qparams.
extra_repr()Return the extra representation of the module.
forward(tensor)Compute scale, zero point, and minval from the input tensor.
get_class(key)get_qparams()Return the computed scale, zero point and minval.
list_registry_keys()list_registry_values()register(key)Register a virtual subclass of an ABC.
set_export_mode([enabled])update_running_range(min_val, max_val)Return
(updated_min, updated_max)using subclass-specific rule.