Enum PrecisionMode 

pub enum PrecisionMode {
    Full,
    Single,
    Half,
    Mixed,
    BFloat16,
    TensorFloat32,
    Auto,
}

Precision mode for GPU computations

Variants

Full

Full precision (f64)

Single

Single precision (f32)

Half

Half precision (f16)

Mixed

Mixed precision (f16 compute, f32 accumulator)

BFloat16

Brain floating point (bf16)

TensorFloat32

Tensor float 32 (TF32) for NVIDIA Ampere+

Auto

Automatic selection based on hardware

Trait Implementations

impl Clone for PrecisionMode

fn clone(&self) -> PrecisionMode

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for PrecisionMode

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for PrecisionMode

fn default() -> Self

Returns the “default value” for a type.

impl<'de> Deserialize<'de> for PrecisionMode

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Display for PrecisionMode

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl PartialEq for PrecisionMode

fn eq(&self, other: &PrecisionMode) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Serialize for PrecisionMode

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl Copy for PrecisionMode

impl Eq for PrecisionMode

impl StructuralPartialEq for PrecisionMode