tf.keras.quantizers.Int4QuantizationConfig

Int4 quantization config.

Inherits From: QuantizationConfig

weight_quantizer Quantizer for weights.
activation_quantizer Quantizer for activations. If "default", uses AbsMaxQuantizer with axis=-1.
block_size Size of groups along the input dimension for sub-channel quantization. If a positive integer, uses sub-channel quantization with ceil(input_dim / block_size) groups. If None or -1, uses per-channel quantization (one scale per output channel). Default: 128 (sub-channel with 128-element groups).

mode

Methods

activation_quantizer_or_default

View source

from_config

View source

get_config

View source

weight_quantizer_or_default

View source