Enum tract_core::ops::cnn::PaddingSpec

pub enum PaddingSpec {
    Explicit(TVec<usize>, TVec<usize>),
    Valid,
    SameUpper,
    SameLower,
}

Variants

Explicit(TVec<usize>, TVec<usize>)

Valid

SameUpper

SameLower

Methods

impl PaddingSpec

pub fn valid_dim(&self, d: usize) -> bool

pub fn rm_axis(&self, d: usize) -> PaddingSpec

pub fn compute<D: DimLike>(
    &self,
    input_spatial_shape: &[D],
    kernel_spatial_shape: &[usize],
    dilations: &[usize],
    strides: &[usize]
) -> TVec<ComputedPaddedDim<D>>

pub fn compute_one<D: DimLike>(
    &self,
    axis: usize,
    input: &D,
    kernel: usize,
    dilation: usize,
    stride: usize
) -> ComputedPaddedDim<D>

Trait Implementations

impl Clone for PaddingSpec

fn clone(&self) -> PaddingSpec

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for PaddingSpec

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

Formats the value using the given formatter.

impl Default for PaddingSpec

fn default() -> PaddingSpec

Returns the "default value" for a type.

impl PartialEq<PaddingSpec> for PaddingSpec

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

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

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

This method tests for !=.

impl StructuralPartialEq for PaddingSpec