Skip to main content

OnnxDataType

uor_addr::onnx::dtype

Enum OnnxDataType

pub enum OnnxDataType {
Show 23 variants    Float,
    Uint8,
    Int8,
    Uint16,
    Int16,
    Int32,
    Int64,
    String,
    Bool,
    Float16,
    Double,
    Uint32,
    Uint64,
    Complex64,
    Complex128,
    Bfloat16,
    Float8E4M3Fn,
    Float8E4M3Fnuz,
    Float8E5M2,
    Float8E5M2Fnuz,
    Uint4,
    Int4,
    Float4E2M1,
}

Expand description

An ONNX tensor element type. All variants except Self::String map 1:1 to a [prism::tensor::dtype] shape.

Variants§

Float

FLOAT (1) → [F32].

Uint8

UINT8 (2) → [U8].

Int8

INT8 (3) → [I8].

Uint16

UINT16 (4) → [U16].

Int16

INT16 (5) → [I16].

Int32

INT32 (6) → [I32].

Int64

INT64 (7) → [I64].

String

STRING (8) — ONNX-specific; not in the numeric dtype alphabet.

Bool

BOOL (9) → [BOOL].

Float16

FLOAT16 (10) → [F16].

Double

DOUBLE (11) → [F64].

Uint32

UINT32 (12) → [U32].

Uint64

UINT64 (13) → [U64].

Complex64

COMPLEX64 (14) → [C64].

Complex128

COMPLEX128 (15) → [C128].

Bfloat16

BFLOAT16 (16) → [BF16].

Float8E4M3Fn

FLOAT8E4M3FN (17) → [F8_E4M3].

Float8E4M3Fnuz

FLOAT8E4M3FNUZ (18) → [F8_E4M3_FNUZ].

Float8E5M2

FLOAT8E5M2 (19) → [F8_E5M2].

Float8E5M2Fnuz

FLOAT8E5M2FNUZ (20) → [F8_E5M2_FNUZ].

Uint4

UINT4 (21) → [U4].

Int4

INT4 (22) → [I4].

Float4E2M1

FLOAT4E2M1 (23) → [F4_E2M1].

Implementations§

impl OnnxDataType

pub const fn from_i32(id: i32) -> Option<Self>

Map a raw TensorProto.DataType ID. Returns None for 0 (UNDEFINED) and IDs outside 1..=23.

pub const fn block_bytes(self) -> Option<usize>

Block bytes from the corresponding [prism::tensor::dtype] shape, or None for Self::String (no fixed element width).

Trait Implementations§

impl Clone for OnnxDataType

fn clone(&self) -> OnnxDataType

Returns a duplicate of the value. Read more

1.0.0 · Source§

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

Performs copy-assignment from source. Read more

impl Debug for OnnxDataType

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

Formats the value using the given formatter. Read more

impl PartialEq for OnnxDataType

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

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

1.0.0 · Source§

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 Copy for OnnxDataType

impl Eq for OnnxDataType

impl StructuralPartialEq for OnnxDataType

Auto Trait Implementations§

impl Freeze for OnnxDataType

impl RefUnwindSafe for OnnxDataType

impl Send for OnnxDataType

impl Sync for OnnxDataType

impl Unpin for OnnxDataType

impl UnsafeUnpin for OnnxDataType

impl UnwindSafe for OnnxDataType

Blanket Implementations§

impl<T> Any for T
where T: 'static + ?Sized,

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more

impl<T> Borrow<T> for T
where T: ?Sized,

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more

impl<T> BorrowMut<T> for T
where T: ?Sized,

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more

impl<T> CloneToUninit for T
where T: Clone,

unsafe fn clone_to_uninit(&self, dest: *mut u8)

🔬This is a nightly-only experimental API. (clone_to_uninit)

Performs copy-assignment from self to dest. Read more

impl<T> From<T> for T

fn from(t: T) -> T

Returns the argument unchanged.

impl<T, U> Into<U> for T
where U: From<T>,

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

impl<T> Same for T

type Output = T

Should always be Self

impl<T> ToOwned for T
where T: Clone,

type Owned = T

The resulting type after obtaining ownership.

fn to_owned(&self) -> T

Creates owned data from borrowed data, usually by cloning. Read more

fn clone_into(&self, target: &mut T)

Uses borrowed data to replace owned data, usually by cloning. Read more

impl<T, U> TryFrom<U> for T
where U: Into<T>,

type Error = Infallible

The type returned in the event of a conversion error.

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.