Enum DType 

#[repr(u32)]
pub enum DType {
    Int8 = 1,
    Uint8 = 0,
    Int16 = 3,
    Uint16 = 2,
    Int32 = 5,
    Uint32 = 4,
    Float16 = 8,
    Float32 = 9,
    Int64 = 7,
}

Quantized or floating-point data type of a model’s logical I/O.

A subset of edgefirst_tensor::DType, numbered and named by it: the discriminant is that enum’s code and the serialized string is its as_str ("i8", "u8", …, "f32"). The older spellings ("int8", "uint8", …, "float32") are still accepted on input; see DType::from_str_code.

Variants

Int8 = 1

Uint8 = 0

Int16 = 3

Uint16 = 2

Int32 = 5

Uint32 = 4

Float16 = 8

Float32 = 9

Int64 = 7

Class maps from an in-graph argmax (TensorFlow/TFLite exports).

Implementations

impl DType

pub fn all() -> &'static [DType]

Every variant, for exhaustive cross-surface tests.

pub const fn code(self) -> u32

The shared code, identical to edgefirst_tensor::DType::code.

pub fn to_tensor_dtype(self) -> DType

The tensor dtype with the same code.

pub fn from_tensor_dtype(dtype: DType) -> Option<Self>

The model I/O dtype for a tensor dtype; None for one a model’s logical I/O does not carry (u64, f64).

pub fn from_code(code: u32) -> Option<Self>

Decode a shared code; None for an unknown code or a tensor dtype a model’s logical I/O does not carry.

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

The string form, identical to edgefirst_tensor::DType::as_str.

pub fn from_str_code(s: &str) -> Option<Self>

Parse the string form, also accepting the older "int8", "uint8", …, "float32" spellings.

pub fn accepted_names() -> impl Iterator<Item = &'static str>

Every accepted input spelling, current and legacy.

impl DType

pub fn size_bytes(self) -> usize

Returns the tensor’s byte width per element.

pub fn is_integer(self) -> bool

Returns true for integer dtypes (quantized tensors).

pub fn is_float(self) -> bool

Returns true for floating-point dtypes.

Trait Implementations

impl Clone for DType

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Copy for DType

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<'de>>(deserializer: D) -> Result<Self, D::Error>

Deserialize this value from the given Serde deserializer.

impl Eq for DType

impl Hash for DType

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

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: &Self) -> bool

Equality operator ==.

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

Inequality operator !=.

impl Serialize for DType

fn serialize<S: Serializer>(&self, serializer: S) -> Result<S::Ok, S::Error>

Serialize this value into the given Serde serializer.

impl StructuralPartialEq for DType