Enum DType

pub enum DType {
    U8,
    U32,
    I16,
    I32,
    I64,
    BF16,
    F16,
    F32,
    F64,
    F8E4M3,
    F6E2M3,
    F6E3M2,
    F4,
    F8E8M0,
}

The different types of elements allowed in tensors.

Variants

U8

U32

I16

I32

I64

BF16

F16

F32

F64

F8E4M3

F6E2M3

6-bit float with 2 exponent bits and 3 mantissa bits (MX6 format)

F6E3M2

6-bit float with 3 exponent bits and 2 mantissa bits (MX6 format)

F4

4-bit float (MX4 format)

F8E8M0

8-bit float with 8 exponent bits and 0 mantissa bits

Implementations

impl DType

pub fn as_str(&self) -> &'static str

String representation for dtypes.

pub fn size_in_bytes(&self) -> usize

The size used by each element in bytes, i.e. 1 for U8, 4 for F32.

pub fn is_int(&self) -> bool

pub fn is_float(&self) -> bool

Trait Implementations

impl Clone for DType

fn clone(&self) -> DType

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for DType

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

Formats the value using the given formatter.

impl From<BnbDType> for DType

fn from(value: BnbDType) -> DType

Converts to this type from the input type.

impl FromStr for DType

type Err = DTypeParseError

The associated error which can be returned from parsing.

fn from_str(s: &str) -> Result<DType, <DType as FromStr>::Err>

Parses a string s to return a value of this type.

impl GetFloatInfo for DType

fn finfo(&self) -> Result<FloatInfo, Error>

impl Hash for DType

fn hash<__H>(&self, state: &mut __H)

where __H: Hasher,

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl PartialEq for DType

fn eq(&self, other: &DType) -> 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 TryFrom<Dtype> for DType

type Error = Error

The type returned in the event of a conversion error.

fn try_from(value: Dtype) -> Result<DType, Error>

Performs the conversion.

impl TryIntoDType for DType

fn try_into_dtype(&self, _: &[&Device]) -> Result<DType, Error>

impl Copy for DType

impl Eq for DType

impl StructuralPartialEq for DType