Skip to main content

policy_from_sensitivity

Function policy_from_sensitivity 

Source
pub fn policy_from_sensitivity(
    scores: &[SensitivityScore],
    threshold: f32,
    default_precision: LayerPrecision,
    sensitive_precision: LayerPrecision,
) -> MixedPrecisionPolicy
Expand description

Build a mixed-precision policy from sensitivity scores.

Layers with sensitivity above threshold are kept at sensitive_precision, while the rest use default_precision.