burn_tensor/tensor/quantization/
calibration.rs

1use crate::{Tensor, backend::Backend};
2
3/// The observed input calibration range.
4#[derive(Clone, Debug)]
5pub struct CalibrationRange<B: Backend> {
6    /// Minimum observed value(s).
7    pub min: Tensor<B, 1>,
8    /// Maximum observed value(s).
9    pub max: Tensor<B, 1>,
10}
11
12/// Calibration method used to compute the quantization range mapping.
13pub enum Calibration {
14    /// Computes quantization range mapping based on the min and max values.
15    MinMax,
16}