Enum tract_core::ops::cnn::conv::KernelFormat

pub enum KernelFormat {
    OIHW,
    HWIO,
}

Variants

OIHW

HWIO

Implementations

impl KernelFormat

pub fn h_axis(&self) -> usize

pub fn spatial_shape<'a, D: DimLike>(&self, full_shape: &'a [D]) -> &'a [D]

pub fn i<'a, D: DimLike>(&self, full_shape: &'a [D]) -> &'a D

pub fn o<'a, D: DimLike>(&self, full_shape: &'a [D]) -> &'a D

pub fn kernel_as_group_o_ihw( &self, kernel: &Tensor, group: usize, input_channels: usize, output_channels: usize ) -> TractResult<Arc<Tensor>>

Trait Implementations

impl Clone for KernelFormat

fn clone(&self) -> KernelFormat

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for KernelFormat

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

Formats the value using the given formatter.

impl Default for KernelFormat

fn default() -> KernelFormat

Returns the “default value” for a type.

impl Hash for KernelFormat

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<KernelFormat> for KernelFormat

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

This method tests for self and other values to be equal, and is used by ==.

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

This method tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Copy for KernelFormat

impl Eq for KernelFormat

impl StructuralEq for KernelFormat

impl StructuralPartialEq for KernelFormat