:orphan: :py:mod:`neural_compressor.tensorflow.keras.quantization.config` ================================================================ .. py:module:: neural_compressor.tensorflow.keras.quantization.config Module Contents --------------- Classes ~~~~~~~ .. autoapisummary:: neural_compressor.tensorflow.keras.quantization.config.StaticQuantConfig Functions ~~~~~~~~~ .. autoapisummary:: neural_compressor.tensorflow.keras.quantization.config.get_all_registered_configs neural_compressor.tensorflow.keras.quantization.config.get_default_static_quant_config .. py:class:: StaticQuantConfig(weight_dtype: str = 'int8', weight_sym: bool = True, weight_granularity: str = 'per_tensor', act_dtype: str = 'int8', act_sym: bool = True, act_granularity: str = 'per_tensor', white_list: Optional[List[neural_compressor.common.base_config.OP_NAME_OR_MODULE_TYPE]] = DEFAULT_WHITE_LIST) Config class for keras static quantization. .. py:function:: get_all_registered_configs() -> Dict[str, neural_compressor.common.base_config.BaseConfig] Get all registered configs for keras framework. .. py:function:: get_default_static_quant_config() -> StaticQuantConfig Generate the default static quant config. :returns: the default keras config.