Struct burn_tensor::quantization::SymmetricQuantization

pub struct SymmetricQuantization<E: Float, Q: PrimInt> {
    pub scale: E,
    /* private fields */
}

Symmetric quantization scheme.

Fields

scale: E

The scaling factor.

Implementations

impl<E: Float, Q: PrimInt> SymmetricQuantization<E, Q>

pub fn init(scale: E) -> Self

Initialize a symmetric quantization scheme with the given parameters.

Trait Implementations

impl<E: Clone + Float, Q: Clone + PrimInt> Clone for SymmetricQuantization<E, Q>

fn clone(&self) -> SymmetricQuantization<E, Q>

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<E: Debug + Float, Q: Debug + PrimInt> Debug for SymmetricQuantization<E, Q>

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

Formats the value using the given formatter.

impl<'de, E, Q: PrimInt> Deserialize<'de> for SymmetricQuantization<E, Q>

where E: Deserialize<'de> + Float,

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl<E: Float, Q: PrimInt> Hash for SymmetricQuantization<E, Q>

fn hash<H: Hasher>(&self, state: &mut H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl<E: Float, Q: PrimInt> PartialEq for SymmetricQuantization<E, Q>

fn eq(&self, other: &Self) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl<E: Float, Q: PrimInt> Quantization<E, Q> for SymmetricQuantization<E, Q>

fn new(alpha: E, beta: E) -> Self

Create a new quantization scheme for an input range [alpha, beta].

fn quantize(&self, values: &[E]) -> Vec<Q>

Convert the values to a lower precision data type.

fn dequantize(&self, values: &[Q]) -> Vec<E>

Convert the values back to a higher precision data type.

impl<E, Q: PrimInt> Serialize for SymmetricQuantization<E, Q>

where E: Serialize + Float,

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl<E: Copy + Float, Q: Copy + PrimInt> Copy for SymmetricQuantization<E, Q>

impl<E: Float, Q: PrimInt> Eq for SymmetricQuantization<E, Q>