Skip to main content

BackendMaxPool2d

Trait BackendMaxPool2d 

Source
pub trait BackendMaxPool2d<N>: Backend<N> {
    // Required methods
    fn max_pool2d(
        &self,
        y: &mut Self::Tensor,
        x: &Self::Tensor,
        conv_info: &Conv2dInfo,
    );
    fn max_pool2d_backprop(
        &self,
        dx: &mut Self::Tensor,
        dy: &Self::Tensor,
        x: &Self::Tensor,
        conv_info: &Conv2dInfo,
    );
}

Required Methods§

Source

fn max_pool2d( &self, y: &mut Self::Tensor, x: &Self::Tensor, conv_info: &Conv2dInfo, )

Source

fn max_pool2d_backprop( &self, dx: &mut Self::Tensor, dy: &Self::Tensor, x: &Self::Tensor, conv_info: &Conv2dInfo, )

Dyn Compatibility§

This trait is dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety".

Implementations on Foreign Types§

Source§

impl<'a, N, T: BackendMaxPool2d<N>> BackendMaxPool2d<N> for &'a T

Source§

fn max_pool2d( &self, y: &mut Self::Tensor, x: &Self::Tensor, conv_info: &Conv2dInfo, )

Source§

fn max_pool2d_backprop( &self, dx: &mut Self::Tensor, dy: &Self::Tensor, x: &Self::Tensor, conv_info: &Conv2dInfo, )

Implementors§