tf.keras.quantizers.TernaryQuantizationConfig

View source on GitHub

Ternary quantization config.

Inherits From: QuantizationConfig

tf.keras.quantizers.TernaryQuantizationConfig()

Quantizes weights to {-1, 0, +1} (BitNet b1.58) and stores them at the information-theoretic floor of log2(3) ~= 1.58 bits/value by packing five trits per byte. The quantization rule (threshold and scale) is owned by the layer, so this config takes no quantizer arguments.

Methods

activation_quantizer_or_default

View source

@staticmethod
activation_quantizer_or_default(
    config, default
)

from_config

View source

@classmethod
from_config(
    config
)

get_config

View source

get_config()

weight_quantizer_or_default

View source

@staticmethod
weight_quantizer_or_default(
    config, default
)