neural_compressor.tensorflow.quantization.autotune
Module Contents
Functions
|
|
|
The main entry of auto-tune. |
- neural_compressor.tensorflow.quantization.autotune.autotune(model: str | tensorflow.keras.Model | neural_compressor.tensorflow.utils.BaseModel, tune_config: neural_compressor.common.base_tuning.TuningConfig, eval_fn: Callable, eval_args: Tuple[Any] | None = None, calib_dataloader: Callable = None, calib_iteration: int = 100) neural_compressor.tensorflow.utils.BaseModel | None [source]
The main entry of auto-tune.