neural_compressor.strategy.utils

Intel Neural Compressor Strategy Utils.

Submodules

• neural_compressor.strategy.utils.tuning_sampler
• neural_compressor.strategy.utils.tuning_space
• neural_compressor.strategy.utils.tuning_structs

Package Contents

Classes

OpTuningConfig

Op tuning config.

class neural_compressor.strategy.utils.OpTuningConfig(op_name, op_type, op_quant_mode, tuning_space, kwargs={})

Op tuning config.

get_state()

Return the op tuning configuration.

Returns:

The op tuning state.

Return type:

Dict

classmethod from_state(config: Dict)

Create the tuning config from dict.

Parameters:

config – A dict includes the tuning config.