Struct tract_hir::ops::cnn::PoolSpec

pub struct PoolSpec {
    pub data_format: DataFormat,
    pub kernel_shape: SmallVec<[usize; 4]>,
    pub padding: PaddingSpec,
    pub dilations: Option<SmallVec<[usize; 4]>>,
    pub strides: Option<SmallVec<[usize; 4]>>,
    pub output_channel_override: Option<usize>,
}

Fields

data_format: DataFormat

kernel_shape: SmallVec<[usize; 4]>

padding: PaddingSpec

dilations: Option<SmallVec<[usize; 4]>>

strides: Option<SmallVec<[usize; 4]>>

output_channel_override: Option<usize>

Implementations

impl PoolSpec

pub fn info(&self) -> Vec<String, Global>

pub fn rank(&self) -> usize

pub fn dilation(&self, geo_axis: usize) -> usize

pub fn dilations(&self) -> Cow<'_, [usize]>

pub fn stride(&self, geo_axis: usize) -> usize

pub fn strides(&self) -> Cow<'_, [usize]>

pub fn output_shape<D>(
    &self,
    input: &[D]
) -> Result<BaseDataShape<D, SmallVec<[D; 4]>>, Error> where
    D: DimLike, 

pub fn output_facts(
    &self,
    inputs: &[&TypedFact]
) -> Result<SmallVec<[TypedFact; 4]>, Error>

pub fn dispose_n_axis(&self) -> PoolSpec

pub fn compute_geo(
    &self,
    input_full_shape: &[TDim]
) -> Result<GeometryBound<SymbolicPoolGeometry, ConcretePoolGeometry>, Error>

impl PoolSpec

pub fn new(
    data_format: DataFormat,
    kernel_shape: SmallVec<[usize; 4]>,
    padding: PaddingSpec,
    dilations: Option<SmallVec<[usize; 4]>>,
    strides: Option<SmallVec<[usize; 4]>>,
    output_channel_override: Option<usize>
) -> PoolSpec

Constructs a new PoolSpec.

Trait Implementations

impl Clone for PoolSpec

pub fn clone(&self) -> PoolSpec

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for PoolSpec

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

Formats the value using the given formatter.

impl Default for PoolSpec

pub fn default() -> PoolSpec

Returns the “default value” for a type.

impl Hash for PoolSpec

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

Feeds a slice of this type into the given Hasher.

impl PartialEq<PoolSpec> for PoolSpec

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

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

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

This method tests for !=.

impl StructuralPartialEq for PoolSpec