pub struct AutoMixedPrecision { /* private fields */ }Expand description
Automatic Mixed Precision (AMP) support
Implementations§
Source§impl AutoMixedPrecision
impl AutoMixedPrecision
pub fn new() -> Self
pub fn enable(self) -> Self
pub fn scale_loss(&mut self, loss: f32) -> f32
pub fn unscale_gradients(&mut self, gradients: &mut [f32])
pub fn update_scale(&mut self, found_inf: bool)
Trait Implementations§
Auto Trait Implementations§
impl Freeze for AutoMixedPrecision
impl RefUnwindSafe for AutoMixedPrecision
impl Send for AutoMixedPrecision
impl Sync for AutoMixedPrecision
impl Unpin for AutoMixedPrecision
impl UnwindSafe for AutoMixedPrecision
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> 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