neural_compressor.ux.utils.workload.mixed_precision

Configuration mixed_precision module.

Module Contents

Classes

MixedPrecision

Configuration Mixed Precision class.

class neural_compressor.ux.utils.workload.mixed_precision.MixedPrecision(data: Dict[str, Any] | None = None)

Bases: neural_compressor.ux.utils.json_serializer.JsonSerializer

Configuration Mixed Precision class.

set_precisions(precisions: str | List[str]) → None

Update mixed_precision precisions in config.