Struct QuantizationParams 

pub struct QuantizationParams { /* private fields */ }

Implementations

impl QuantizationParams

pub fn new(scale: f64, zero_point: i64) -> Result<Self, TensorMetaError>

Create affine quantization parameters.

scale must be finite and strictly positive. It is stored by bit pattern so tensor metadata remains equality/hash stable.

pub fn per_channel( scales: Vec<f64>, zero_points: Vec<i64>, axis: usize, ) -> Result<Self, TensorMetaError>

Create per-channel affine quantization parameters.

axis identifies the quantized channel dimension in the tensor shape. scales and zero_points must have the same non-zero length.

pub fn scale(&self) -> f64

pub fn zero_point(&self) -> i64

pub fn scales(&self) -> Vec<f64>

pub fn zero_points(&self) -> &[i64]

pub fn axis(&self) -> Option<usize>

pub fn len(&self) -> usize

pub fn is_empty(&self) -> bool

Trait Implementations

impl Clone for QuantizationParams

fn clone(&self) -> QuantizationParams

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for QuantizationParams

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

Formats the value using the given formatter.

impl Eq for QuantizationParams

impl Hash for QuantizationParams

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 PartialEq for QuantizationParams

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

Equality operator ==.

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

Inequality operator !=.

impl StructuralPartialEq for QuantizationParams