tfmot.quantization.keras.default_8bit.Default8BitQuantizeRegistry

QuantizationRegistry for built-in Keras classes for default 8-bit scheme.

Inherits From: QuantizeRegistry

tfmot.quantization.keras.default_8bit.Default8BitQuantizeRegistry(
    disable_per_axis=False
)

Methods

get_quantize_config

View source

get_quantize_config(
    layer
)

Returns the quantization config for the given layer.

Args
layer input layer to return quantize config for.

Returns
Returns the QuantizeConfig for the given layer.

supports

View source

supports(
    layer
)

Returns whether the registry supports this layer type.

Args
layer The layer to check for support.

Returns
True/False whether the layer type is supported.