Struct tract_core::ops::cnn::pools::PoolSpec

pub struct PoolSpec {
    pub data_format: DataFormat,
    pub kernel_shape: TVec<usize>,
    pub padding: PaddingSpec,
    pub dilations: Option<TVec<usize>>,
    pub strides: Option<TVec<usize>>,
    pub output_channel_override: Option<usize>,
}

Fields

data_format: DataFormat

kernel_shape: TVec<usize>

padding: PaddingSpec

dilations: Option<TVec<usize>>

strides: Option<TVec<usize>>

output_channel_override: Option<usize>

Methods

impl PoolSpec

pub fn new(
    data_format: DataFormat,
    kernel_shape: TVec<usize>,
    padding: PaddingSpec,
    dilations: Option<TVec<usize>>,
    strides: Option<TVec<usize>>,
    output_channel_override: Option<usize>
) -> Self

Constructs a new PoolSpec.

impl PoolSpec

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

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

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

pub fn compute_geo(
    &self,
    input_full_shape: &[usize]
) -> TractResult<(DataShape, Patch, DataShape)>

pub fn output_facts(
    &self,
    inputs: &[&TypedFact]
) -> TractResult<TVec<TypedFact>>

pub fn pulsify(
    &self,
    _source: &NormalizedModel,
    node: &NormalizedNode,
    op: &dyn PulsedOp,
    target: &mut PulsedModel,
    mapping: &HashMap<OutletId, OutletId>
) -> TractResult<TVec<OutletId>>

pub fn dispose_n_axis(&self) -> PoolSpec

pub fn pulsed_output_facts(
    &self,
    inputs: &[&PulsedFact]
) -> TractResult<TVec<PulsedFact>>

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

Formats the value using the given formatter.

impl Default for PoolSpec

fn default() -> PoolSpec

Returns the "default value" for a type.