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 Freeze for MathType
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