pub struct TensorCoresConfig {
pub enabled: bool,
pub data_type: TensorDataType,
pub tile_size: (usize, usize, usize),
pub mixed_precision: bool,
pub loss_scale: f32,
pub auto_mixed_precision: bool,
pub min_matrix_size: usize,
}Expand description
Tensor cores optimization configuration
Fields§
§enabled: boolEnable tensor cores acceleration
data_type: TensorDataTypePreferred data type for computation
tile_size: (usize, usize, usize)Matrix dimensions to use for tiling
mixed_precision: boolEnable mixed precision training
loss_scale: f32Loss scaling for mixed precision
auto_mixed_precision: boolEnable automatic mixed precision
min_matrix_size: usizeMinimum matrix size to use tensor cores
Trait Implementations§
Source§impl Clone for TensorCoresConfig
impl Clone for TensorCoresConfig
Source§fn clone(&self) -> TensorCoresConfig
fn clone(&self) -> TensorCoresConfig
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 moreSource§impl Debug for TensorCoresConfig
impl Debug for TensorCoresConfig
Auto Trait Implementations§
impl Freeze for TensorCoresConfig
impl RefUnwindSafe for TensorCoresConfig
impl Send for TensorCoresConfig
impl Sync for TensorCoresConfig
impl Unpin for TensorCoresConfig
impl UnwindSafe for TensorCoresConfig
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 moreSource§impl<T> Pointable for T
impl<T> Pointable for T
Source§impl<SS, SP> SupersetOf<SS> for SPwhere
SS: SubsetOf<SP>,
impl<SS, SP> SupersetOf<SS> for SPwhere
SS: SubsetOf<SP>,
Source§fn to_subset(&self) -> Option<SS>
fn to_subset(&self) -> Option<SS>
The inverse inclusion map: attempts to construct
self from the equivalent element of its
superset. Read moreSource§fn is_in_subset(&self) -> bool
fn is_in_subset(&self) -> bool
Checks if
self is actually part of its subset T (and can be converted to it).Source§fn to_subset_unchecked(&self) -> SS
fn to_subset_unchecked(&self) -> SS
Use with care! Same as
self.to_subset but without any property checks. Always succeeds.Source§fn from_subset(element: &SS) -> SP
fn from_subset(element: &SS) -> SP
The inclusion map: converts
self to the equivalent element of its superset.