Struct ActivationQuantizer 

pub struct ActivationQuantizer { /* private fields */ }

Activation quantization manager

Implementations

impl ActivationQuantizer

pub fn new(config: ActivationQuantConfig) -> Self

Create new activation quantizer

pub fn start_calibration(&mut self)

Start calibration phase

pub fn end_calibration(&mut self)

End calibration phase

pub fn is_calibration_complete(&self) -> bool

Check if calibration is complete

pub fn quantize_activation( &mut self, tensor: &Tensor, layer_name: &str, training: bool, ) -> Result<Tensor>

Quantize activation tensor for a specific layer

pub fn get_layer_stats(&self, layer_name: &str) -> Option<&ActivationStats>

Get statistics for a specific layer

pub fn get_all_stats(&self) -> &HashMap<String, ActivationStats>

Get all layer statistics

pub fn save_calibration(&self, path: &str) -> Result<()>

Save calibration statistics to file

pub fn load_calibration(&mut self, path: &str) -> Result<()>

Load calibration statistics from file

pub fn configure_layer(&mut self, layer_name: &str, config: LayerQuantConfig)

Configure quantization for a specific layer

pub fn disable_layer(&mut self, layer_name: &str)

Disable quantization for a specific layer

pub fn get_memory_savings(&self) -> f32

Get memory savings from activation quantization