Enum DataType 

#[repr(i32)]
pub enum DataType {
    kFLOAT = 0,
    kHALF = 1,
    kINT8 = 2,
    kINT32 = 3,
    kBOOL = 4,
    kUINT8 = 5,
    kFP8 = 6,
    kBF16 = 7,
    kINT64 = 8,
    kINT4 = 9,
    kFP4 = 10,
    kE8M0 = 11,
}

! ! \enum DataType ! \brief The type of weights and tensors. ! The datatypes other than kBOOL, kINT32, and kINT64 are “activation datatypes,” ! as they often represent values corresponding to inference results. !

Variants

kFLOAT = 0

! 32-bit floating point format.

kHALF = 1

! IEEE 16-bit floating-point format – has a 5 bit exponent and 11 bit significand.

kINT8 = 2

! Signed 8-bit integer representing a quantized floating-point value.

kINT32 = 3

! Signed 32-bit integer format.

kBOOL = 4

! 8-bit boolean. 0 = false, 1 = true, other values undefined.

kUINT8 = 5

! Unsigned 8-bit integer format. ! Cannot be used to represent quantized floating-point values. ! Use the IdentityLayer to convert kUINT8 network-level inputs to {kFLOAT, kHALF} prior ! to use with other TensorRT layers, or to convert intermediate output ! before kUINT8 network-level outputs from {kFLOAT, kHALF} to kUINT8. ! kUINT8 conversions are only supported for {kFLOAT, kHALF}. ! kUINT8 to {kFLOAT, kHALF} conversion will convert the integer values ! to equivalent floating point values. ! {kFLOAT, kHALF} to kUINT8 conversion will convert the floating point values ! to integer values by truncating towards zero. This conversion has undefined behavior for ! floating point values outside the range [0.0F, 256.0F) after truncation. ! kUINT8 conversions are not supported for {kINT8, kINT32, kBOOL}.

kFP8 = 6

! Signed 8-bit floating point with ! 1 sign bit, 4 exponent bits, 3 mantissa bits, and exponent-bias 7.

kBF16 = 7

! Brain float – has an 8 bit exponent and 8 bit significand.

kINT64 = 8

! Signed 64-bit integer type.

kINT4 = 9

! Signed 4-bit integer type.

kFP4 = 10

! 4-bit floating point type ! 1 bit sign, 2 bit exponent, 1 bit mantissa

kE8M0 = 11

! Unsigned representation of exponent-only 8-bit floating point type for quantization scales

Implementations

impl DataType

pub const fn size_bits(self) -> usize

Trait Implementations

impl Clone for DataType

fn clone(&self) -> DataType

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DataType

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

Formats the value using the given formatter.

impl From<DataType> for DataType

fn from(value: DataType) -> Self

Converts to this type from the input type.

impl Hash for DataType

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 Into<DataType> for DataType

fn into(self) -> DataType

Converts this type into the (usually inferred) input type.

impl Ord for DataType

fn cmp(&self, other: &DataType) -> 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,

Restrict a value to a certain interval.

impl PartialEq for DataType

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

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

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

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

Tests less than (for self and other) and is used by the < operator.

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

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

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

Tests greater than (for self and other) and is used by the > operator.

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

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

impl Copy for DataType

impl Eq for DataType

impl StructuralPartialEq for DataType