burn_backend/tensor/quantization/
parameters.rs

1use crate::Backend;
2
3pub use burn_std::quantization::{QParamTensor, QParams};
4
5/// The quantization parameters primitive.
6///
7/// # Remarks
8///
9/// This is a low-level struct used internally by the library to provide the quantization parameters
10/// to the backends. It is not designed for direct usage by users, and not recommended to import
11/// or use this struct directly.
12pub struct QuantizationParametersPrimitive<B: Backend> {
13    /// The scaling factor.
14    pub scales: B::FloatTensorPrimitive,
15}