qualia_codegen_core.Quantizer module

class qualia_codegen_core.Quantizer.Quantizer(width: int)[source]

Bases: object

quantize_array_with_scale_factor(arr: ndarray[Any, dtype[floating[Any]]] | ndarray[Any, dtype[integer[Any]]], scale_factor: int, round_mode: RoundMode) → ndarray[Any, dtype[floating[Any]]] | ndarray[Any, dtype[integer[Any]]] | None[source]

quantize_weights_with_scale_factor(node: LayerNode, scale_factor: int, round_mode: RoundMode, bias_scale_factor: int | None = None, exclude: list[str] | None = None) → bool[source]

quantize_weights(node: LayerNode, exclude: list[str] | None = None) → bool[source]