Enum coaster_nn::MathType
source · pub enum MathType {
Default,
TensorOPMath,
TensorOPMathAllowConversion,
}
Expand description
Indicate if Tensor Core Operations are permitted [cudnnMathType_t][1] [1]: https://docs.nvidia.com/deeplearning/sdk/cudnn-api/index.html#cudnnMathType_t
Variants§
Default
No Tensor Core ops
TensorOPMath
Uses Tensor Core ops
TensorOPMathAllowConversion
Uses FP32 Tensors for input/output
Trait Implementations§
Auto Trait Implementations§
impl RefUnwindSafe for MathType
impl Send for MathType
impl Sync for MathType
impl Unpin for MathType
impl UnwindSafe for MathType
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