Struct 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 input_channels: usize,
    pub output_channels: usize,
}

Fields

data_format: DataFormat

kernel_shape: SmallVec<[usize; 4]>

padding: PaddingSpec

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

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

input_channels: usize

output_channels: usize

Implementations

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]>>, input_channels: usize, output_channels: usize, ) -> PoolSpec

Constructs a new PoolSpec.

impl PoolSpec

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

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 computed_padding<D>( &self, input_hw: &[D], ) -> SmallVec<[ComputedPaddedDim<D>; 4]>

where D: DimLike,

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>

pub fn change_geo_axes(&self, op: &AxisOp) -> Result<PoolSpec, Error>

pub fn declutter(&self, input: &[TDim]) -> Result<Option<PoolSpec>, Error>

Trait Implementations

impl Clone for PoolSpec

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

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

Formats the value using the given formatter.

impl Default for PoolSpec

fn default() -> PoolSpec

Returns the “default value” for a type.

impl Hash for PoolSpec

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

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

impl StructuralPartialEq for PoolSpec