neural_compressor.tensorflow.algorithms.static_quant.keras

Module Contents

Classes

KerasAdaptor	The keras class of framework adaptor layer.
KerasQuery	Class that queries configs from yaml settings.
KerasConfigConverter	Convert StaticQuantConfig to the format used by static quant algo.
KerasSurgery	The class that inserts FakeQuant or QDQ layers before the target layers.

class neural_compressor.tensorflow.algorithms.static_quant.keras.KerasAdaptor(framework_specific_info)

The keras class of framework adaptor layer.

class neural_compressor.tensorflow.algorithms.static_quant.keras.KerasQuery(local_config_file=None)

Class that queries configs from yaml settings.

class neural_compressor.tensorflow.algorithms.static_quant.keras.KerasConfigConverter(quant_config: neural_compressor.tensorflow.quantization.config.StaticQuantConfig, calib_iteration: int)

Convert StaticQuantConfig to the format used by static quant algo.

class neural_compressor.tensorflow.algorithms.static_quant.keras.KerasSurgery(model)

The class that inserts FakeQuant or QDQ layers before the target layers.