[−][src]Trait yarnn::backend::BackendConv2d

pub trait BackendConv2d<N>: Backend<N> {
    type Context;
    fn conv2d_forward(
        &self, 
        y: &mut Self::Tensor, 
        x: &Self::Tensor, 
        filter: &Self::Tensor, 
        conv_info: &Conv2dInfo
    );

    fn conv2d_backward_input(
        &self, 
        dx: &mut Self::Tensor, 
        dy: &Self::Tensor, 
        filter: &Self::Tensor, 
        conv_info: &Conv2dInfo
    );

    fn conv2d_backward_filter(
        &self, 
        dw: &mut Self::Tensor, 
        x: &Self::Tensor, 
        dy: &Self::Tensor, 
        conv_info: &Conv2dInfo
    );
}

Associated Types

type Context

Loading content...

Required methods

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

fn conv2d_backward_input(
    &self,
    dx: &mut Self::Tensor,
    dy: &Self::Tensor,
    filter: &Self::Tensor,
    conv_info: &Conv2dInfo
)

fn conv2d_backward_filter(
    &self,
    dw: &mut Self::Tensor,
    x: &Self::Tensor,
    dy: &Self::Tensor,
    conv_info: &Conv2dInfo
)

Loading content...

Implementations on Foreign Types

impl<'a, N, T: BackendConv2d<N>> BackendConv2d<N> for &'a T[src]

type Context = ()

Loading content...

Implementors

impl BackendConv2d<f32> for Native[src]

type Context = ()

Loading content...