Enum safetensors::tensor::Dtype

#[non_exhaustive]
pub enum Dtype {
    BOOL,
    U8,
    I8,
    F8_E5M2,
    F8_E4M3,
    I16,
    U16,
    F16,
    BF16,
    I32,
    U32,
    F32,
    F64,
    I64,
    U64,
}

The various available dtypes. They MUST be in increasing alignment order

Variants (Non-exhaustive)

Non-exhaustive enums could have additional variants added in future. Therefore, when matching against variants of non-exhaustive enums, an extra wildcard arm must be added to account for any future variants.

BOOL

Boolan type

U8

Unsigned byte

I8

Signed byte

F8_E5M2

FP8 https://arxiv.org/pdf/2209.05433.pdf_

F8_E4M3

FP8 https://arxiv.org/pdf/2209.05433.pdf_

I16

Signed integer (16-bit)

U16

Unsigned integer (16-bit)

F16

Half-precision floating point

BF16

Brain floating point

I32

Signed integer (32-bit)

U32

Unsigned integer (32-bit)

F32

Floating point (32-bit)

F64

Floating point (64-bit)

I64

Signed integer (64-bit)

U64

Unsigned integer (64-bit)

Implementations

impl Dtype

pub fn size(&self) -> usize

Gives out the size (in bytes) of 1 element of this dtype.

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for Dtype

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

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Ord for Dtype

fn cmp(&self, other: &Dtype) -> Ordering

This method returns an Ordering between self and other.

fn max(self, other: Self) -> Self

where Self: Sized,

Compares and returns the maximum of two values.

fn min(self, other: Self) -> Self

where Self: Sized,

Compares and returns the minimum of two values.

fn clamp(self, min: Self, max: Self) -> Self

where Self: Sized + PartialOrd,

Restrict a value to a certain interval.

impl PartialEq for Dtype

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

This method tests for self and other values to be equal, and is used by ==.

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

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

impl PartialOrd for Dtype

fn partial_cmp(&self, other: &Dtype) -> Option<Ordering>

This method returns an ordering between self and other values if one exists.

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

This method tests less than (for self and other) and is used by the < operator.

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

This method tests less than or equal to (for self and other) and is used by the <= operator.

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

This method tests greater than (for self and other) and is used by the > operator.

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

This method tests greater than or equal to (for self and other) and is used by the >= operator.

impl Serialize for Dtype

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 Dtype

impl Eq for Dtype

impl StructuralPartialEq for Dtype