tfmot.quantization.keras.experimental.default_n_bit.DefaultNBitQuantizeRegistry

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

Inherits From: QuantizeRegistry

Methods

get_quantize_config

View source

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

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.