Struct SensitivityAnalyzer 

pub struct SensitivityAnalyzer { /* private fields */ }

Quantization sensitivity analyzer

Implementations

impl SensitivityAnalyzer

pub fn new(test_data: Vec<(Tensor, Tensor)>) -> Self

Create a new sensitivity analyzer

pub fn set_tolerance(&mut self, tolerance: f32)

Set tolerance for accuracy comparison

pub fn analyze_layer_sensitivity( &self, layer_names: &[String], evaluation_fn: impl Fn(&str, &QuantConfig) -> TorshResult<f32>, ) -> TorshResult<SensitivityAnalysisResults>

Perform sensitivity analysis on a model’s layers

pub fn heuristic_sensitivity_analysis( &self, layer_names: &[String], ) -> TorshResult<SensitivityAnalysisResults>

Perform heuristic sensitivity analysis based on layer types

pub fn compare_tensor_accuracy( &self, original: &Tensor, quantized: &Tensor, ) -> TorshResult<f32>

Compare accuracy between original and quantized tensors

pub fn calculate_mse( &self, original: &Tensor, quantized: &Tensor, ) -> TorshResult<f32>

Calculate Mean Squared Error between tensors

pub fn calculate_snr( &self, original: &Tensor, quantized: &Tensor, ) -> TorshResult<f32>

Calculate Signal-to-Noise Ratio