pub enum MixedPrecisionArray {
Full(Array2<Float>),
Half(Array2<Half>),
Mixed {
fp32_data: Array2<Float>,
fp16_data: Array2<Half>,
precision_mask: Array2<bool>,
},
}Expand description
Mixed precision data types
Variants§
Full(Array2<Float>)
Full precision (FP32)
Half(Array2<Half>)
Half precision (FP16)
Mixed
Mixed arrays with different precisions per operation
Implementations§
Source§impl MixedPrecisionArray
impl MixedPrecisionArray
Sourcepub fn is_mixed_precision(&self) -> bool
pub fn is_mixed_precision(&self) -> bool
Check if array uses mixed precision
Sourcepub fn memory_usage_bytes(&self) -> usize
pub fn memory_usage_bytes(&self) -> usize
Get memory usage in bytes
Trait Implementations§
Source§impl Clone for MixedPrecisionArray
impl Clone for MixedPrecisionArray
Source§fn clone(&self) -> MixedPrecisionArray
fn clone(&self) -> MixedPrecisionArray
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 MixedPrecisionArray
impl RefUnwindSafe for MixedPrecisionArray
impl Send for MixedPrecisionArray
impl Sync for MixedPrecisionArray
impl Unpin for MixedPrecisionArray
impl UnwindSafe for MixedPrecisionArray
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> 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