neural_compressor.strategy.utils.tuning_structs¶
Tuning structure.
Module Contents¶
Classes¶
Op tuning config. |
- class neural_compressor.strategy.utils.tuning_structs.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.