Enum KernelFormat

pub enum KernelFormat {
    OIHW,
    HWIO,
    OHWI,
}

Variants

OIHW

HWIO

OHWI

Implementations

impl KernelFormat

pub fn h_axis(&self) -> usize

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

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

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

pub fn o_axis<D>(&self, full_shape: &[D]) -> usize

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

pub fn input_channels<'s, D: DimLike>( &self, full_kernel_shape: &'s [D], group: usize, ) -> Cow<'s, D>

pub fn output_channels<'s, D: DimLike>( &self, full_kernel_shape: &'s [D], group: usize, ) -> Cow<'s, D>

pub fn kernel_as_group_o_i_h_w_ops( &self, full_shape: &[impl DimLike], group: usize, ) -> TVec<AxisOp>

pub fn kernel_as_group_o_i_hw_ops( &self, full_shape: &[impl DimLike], group: usize, ) -> TVec<AxisOp>

pub fn kernel_as_group_o_ihw_ops( &self, full_shape: &[impl DimLike], group: usize, ) -> TVec<AxisOp>

pub fn kernel_as_group_o_i_hw( &self, kernel: &Tensor, group: usize, ) -> TractResult<Tensor>

pub fn kernel_as_group_o_ihw( &self, kernel: &Tensor, group: usize, ) -> TractResult<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 for KernelFormat

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

impl Eq for KernelFormat

impl StructuralPartialEq for KernelFormat