Struct OperatorMaxPool 

pub struct OperatorMaxPool {
    pub auto_pad: Option<AutoPad>,
    pub ceil_mode: Option<bool>,
    pub dilations: Option<Vec<usize>>,
    pub kernel_shape: Vec<usize>,
    pub pads: Option<Vec<usize>>,
    pub storage_order: Option<usize>,
    pub strides: Option<Vec<usize>>,
}

Fields

auto_pad: Option<AutoPad>

ceil_mode: Option<bool>

dilations: Option<Vec<usize>>

kernel_shape: Vec<usize>

pads: Option<Vec<usize>>

storage_order: Option<usize>

strides: Option<Vec<usize>>

Implementations

impl OperatorMaxPool

pub fn new(kernel_shape: &[usize]) -> Self

impl OperatorMaxPool

pub fn create_outputs( &self, inputs: &[Option<Tensor>], expand_ratio: f64, rng: &mut impl RngExt, ) -> Result<Vec<Option<Tensor>>, SimError>

pub fn create_inputs( &self, outputs: &[Option<Tensor>], _expand_ratio: f64, _rng: &mut impl RngExt, ) -> Result<Vec<Option<Tensor>>, SimError>

Trait Implementations

impl Clone for OperatorMaxPool

fn clone(&self) -> OperatorMaxPool

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for OperatorMaxPool

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

Formats the value using the given formatter.

impl Default for OperatorMaxPool

fn default() -> Self

Returns the “default value” for a type.

impl<'de> Deserialize<'de> for OperatorMaxPool

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Eq for OperatorMaxPool

impl Operator for OperatorMaxPool

fn validate_tensors( &self, inputs: &[Option<Tensor>], outputs: &[Option<Tensor>], ) -> SimResult

Validate that the input and output tensors are valid shapes and datatypes

fn compute_delay_ticks( &self, compute_capabilities: &Rc<ComputeCapabilities>, inputs: &[Option<TensorView>], outputs: &[Option<TensorView>], ) -> Result<usize, SimError>

Returns the number of clock ticks needed to perform the specified computation give the machine capabilities

fn compute_machine_ops( &self, inputs: &[Option<TensorView>], outputs: &[Option<TensorView>], ) -> Result<MachineOpCounts, SimError>

Returns the number of machine operations performed by the specified computation, broken down by operation type.

fn partition_views( &self, input_views: &[Option<TensorView>], output_views: &[Option<TensorView>], num_partitions: usize, ) -> Result<Vec<TensorPartition>, SimError>

Partition the operation into one or more views that can be executed in parallel. Implementations may return fewer than num_partitions if the operator cannot be split that finely.

fn compute_flops( &self, inputs: &[Option<TensorView>], outputs: &[Option<TensorView>], ) -> Result<usize, SimError>

Returns the total number of FLOPs performed by the specified computation.

impl PartialEq for OperatorMaxPool

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

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl StructuralPartialEq for OperatorMaxPool