Enum tract_core::ops::cnn::PaddingSpec

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

Variants

Explicit(TVec<usize>, TVec<usize>, 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: DimLike>( &self, input_spatial_shape: &[D], kernel_spatial_shape: &[usize], dilations: &[usize], strides: &[usize] ) -> TVec<ComputedPaddedDim<D>>

pub fn compute_for_deconv<D: DimLike>( &self, conv_spatial_shape: &[D], kernel_spatial_shape: &[usize], dilations: &[usize], strides: &[usize], adjustments: &[usize] ) -> TractResult<TVec<ComputedPaddedDim<D>>>

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

pub fn compute_one_for_deconv<D: DimLike>( &self, axis: usize, input: &D, kernel: usize, dilation: usize, stride: usize, adjustment: usize ) -> TractResult<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 Hash for PaddingSpec

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