tfmot.quantization.keras.experimental.default_n_bit.DefaultNBitQuantizeScheme

Default N-Bit Scheme supported by TFLite.

Inherits From: QuantizeScheme

tfmot.quantization.keras.experimental.default_n_bit.DefaultNBitQuantizeScheme(
    disable_per_axis=False, num_bits_weight=8, num_bits_activation=8
)

Methods

get_layout_transformer

View source

get_layout_transformer()

Returns the layout transforms for this scheme.

Returns
Returns the QuantizeLayoutTransform for this quantization scheme.

get_quantize_registry

View source

get_quantize_registry()

Returns the quantization registry for this scheme.

Returns
Returns the QuantizeRegistry for this quantization scheme.