Enum DataFormat 

pub enum DataFormat {
    Nhwc,
    Nchw,
    Nc4hw4,
}

Data format for tensors.

Variants

Nhwc

NHWC format: (batch, height, width, channels) Common in TensorFlow models

Nchw

NCHW format: (batch, channels, height, width) Common in PyTorch/ONNX models

Nc4hw4

NC4HW4 format: Optimized format for GPU backends

Implementations

impl DataFormat

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

Get the name of this format

Trait Implementations

impl Clone for DataFormat

fn clone(&self) -> DataFormat

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DataFormat

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

Formats the value using the given formatter.

impl Default for DataFormat

fn default() -> DataFormat

Returns the “default value” for a type.

impl Hash for DataFormat

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 DataFormat

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

impl Eq for DataFormat

impl StructuralPartialEq for DataFormat