Enum tract_pulse::internal::tract_core::ops::cnn::PaddingSpec

pub enum PaddingSpec {
    Explicit(SmallVec<[usize; 4]>, SmallVec<[usize; 4]>, bool),
    Valid,
    SameUpper,
    SameLower,
}

Variants

Explicit(SmallVec<[usize; 4]>, SmallVec<[usize; 4]>, bool)

Valid

SameUpper

SameLower

Implementations

impl PaddingSpec

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

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

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

pub fn compute_for_deconv<D>( &self, conv_spatial_shape: &[D], kernel_spatial_shape: &[usize], dilations: &[usize], strides: &[usize], adjustments: &[usize] ) -> Result<SmallVec<[ComputedPaddedDim<D>; 4]>, Error>where D: DimLike,

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

pub fn compute_one_for_deconv<D>( &self, axis: usize, input: &D, kernel: usize, dilation: usize, stride: usize, adjustment: usize ) -> Result<ComputedPaddedDim<D>, Error>where D: DimLike,

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<(), Error>

Formats the value using the given formatter.

impl Default for PaddingSpec

fn default() -> PaddingSpec

Returns the “default value” for a type.

impl Hash for PaddingSpec

fn hash<__H>(&self, state: &mut __H)where __H: Hasher,

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<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: &Rhs) -> bool

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

impl Eq for PaddingSpec

impl StructuralEq for PaddingSpec

impl StructuralPartialEq for PaddingSpec