Module: tfmot.quantization.keras

View source on GitHub

Module containing quantization code built on Keras abstractions.


quantizers module: Module containing Quantization abstraction and quantizers.


class QuantizeConfig: ABC interface for Keras layers to express how they should be quantized.


quantize_annotate_layer(...): Annotate a tf.keras layer to be quantized.

quantize_annotate_model(...): Annotate a tf.keras model to be quantized.

quantize_apply(...): Quantize a tf.keras model that has been annotated for quantization.

quantize_model(...): Quantize a tf.keras model with the default quantization implementation.

quantize_scope(...): Scope which can be used to deserialize quantized Keras models and layers.