Struct HistogramCollector 

pub struct HistogramCollector {
    pub num_bins: usize,
    pub histograms: Vec<(String, TensorHistogram)>,
}

Histogram collector that aggregates activation values across calibration samples.

Fields

num_bins: usize

Number of bins for histograms.

histograms: Vec<(String, TensorHistogram)>

Collected histograms, keyed by tensor name.

Implementations

impl HistogramCollector

pub fn with_bins(num_bins: usize) -> Self

Create a collector with a custom number of bins.

pub fn add_values( &mut self, name: &str, values: &[f32], ) -> Result<(), CalibrationError>

Add values for a named tensor. If a histogram already exists for this tensor, the new values are merged into it.

pub fn collect_from_dataset( &mut self, dataset: &CalibrationDataset, tensor_name: &str, ) -> Result<(), CalibrationError>

Collect histograms from a calibration dataset.

Each sample is treated as a single tensor named tensor_name.

pub fn get(&self, name: &str) -> Option<&TensorHistogram>

Get the histogram for a named tensor.

Trait Implementations

impl Clone for HistogramCollector

fn clone(&self) -> HistogramCollector

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for HistogramCollector

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for HistogramCollector

fn default() -> Self

Returns the “default value” for a type.