Struct DLDataType 

#[repr(C)]
pub struct DLDataType {
    pub code: u8,
    pub bits: u8,
    pub lanes: u16,
}

Data type descriptor specifying the element type of a tensor.

This corresponds to DLDataType in the DLPack specification.

Fields

code: u8

Type code (signed int, unsigned int, float, etc.)

bits: u8

Number of bits per element (e.g., 32 for float32)

lanes: u16

Number of lanes for vectorized types (usually 1)

Implementations

impl DLDataType

pub fn new(code: DLDataTypeCode, bits: u8, lanes: u16) -> Self

Create a new data type descriptor.

pub fn code_enum(&self) -> Option<DLDataTypeCode>

Get the type code as an enum.

Returns None for unknown type codes.

pub fn is_f16(&self) -> bool

Check if this is f16 (half precision float).

pub fn is_f32(&self) -> bool

Check if this is f32 (single precision float).

pub fn is_f64(&self) -> bool

Check if this is f64 (double precision float).

pub fn is_bf16(&self) -> bool

Check if this is bf16 (bfloat16).

pub fn is_i8(&self) -> bool

Check if this is i8 (signed 8-bit integer).

pub fn is_i16(&self) -> bool

Check if this is i16 (signed 16-bit integer).

pub fn is_i32(&self) -> bool

Check if this is i32 (signed 32-bit integer).

pub fn is_i64(&self) -> bool

Check if this is i64 (signed 64-bit integer).

pub fn is_u8(&self) -> bool

Check if this is u8 (unsigned 8-bit integer).

pub fn is_u16(&self) -> bool

Check if this is u16 (unsigned 16-bit integer).

pub fn is_u32(&self) -> bool

Check if this is u32 (unsigned 32-bit integer).

pub fn is_u64(&self) -> bool

Check if this is u64 (unsigned 64-bit integer).

pub fn is_bool(&self) -> bool

Check if this is bool.

pub fn itemsize(&self) -> usize

Get the size of one element in bytes.

Trait Implementations

impl Clone for DLDataType

fn clone(&self) -> DLDataType

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DLDataType

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

Formats the value using the given formatter.

impl Hash for DLDataType

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 DLDataType

fn eq(&self, other: &DLDataType) -> 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 Copy for DLDataType

impl Eq for DLDataType

impl StructuralPartialEq for DLDataType