pub struct QuantizedWeight {
pub data: Vec<i8>,
pub shape: Vec<usize>,
pub params: QuantizationParams,
}Expand description
Quantized weight tensor
Fields§
§data: Vec<i8>Quantized data (INT8)
shape: Vec<usize>Original shape
params: QuantizationParamsQuantization parameters
Implementations§
Source§impl QuantizedWeight
impl QuantizedWeight
Sourcepub fn new(
data: Vec<i8>,
shape: Vec<usize>,
params: QuantizationParams,
) -> ModelResult<Self>
pub fn new( data: Vec<i8>, shape: Vec<usize>, params: QuantizationParams, ) -> ModelResult<Self>
Create a new quantized weight
Sourcepub fn dequantize_1d(&self) -> ModelResult<Array1<f32>>
pub fn dequantize_1d(&self) -> ModelResult<Array1<f32>>
Dequantize to f32 array (1D)
Sourcepub fn dequantize_2d(&self) -> ModelResult<Array2<f32>>
pub fn dequantize_2d(&self) -> ModelResult<Array2<f32>>
Dequantize to f32 array (2D)
Sourcepub fn memory_size(&self) -> usize
pub fn memory_size(&self) -> usize
Get memory size in bytes
Trait Implementations§
Source§impl Clone for QuantizedWeight
impl Clone for QuantizedWeight
Source§fn clone(&self) -> QuantizedWeight
fn clone(&self) -> QuantizedWeight
Returns a duplicate of the value. Read more
1.0.0 · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreAuto Trait Implementations§
impl Freeze for QuantizedWeight
impl RefUnwindSafe for QuantizedWeight
impl Send for QuantizedWeight
impl Sync for QuantizedWeight
impl Unpin for QuantizedWeight
impl UnwindSafe for QuantizedWeight
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more