Trait coaster_nn::NN

pub trait NN<F> {
    type CC: NNOperationConfig<F> + ConvolutionConfig<F>;
    type CLRN: NNOperationConfig<F>;
    type CPOOL: NNOperationConfig<F>;
    type CDROP: NNOperationConfig<F>;
    fn init_nn();
}

Provides the functionality for a backend to support Neural Network related operations.

Associated Types

type CC: NNOperationConfig<F> + ConvolutionConfig<F>

The Convolution Operation Config representation for this Plugin.

type CLRN: NNOperationConfig<F>

The LRN Operation Config representation for this Plugin.

type CPOOL: NNOperationConfig<F>

The Pooling Operation Config representation for this Plugin.

type CDROP: NNOperationConfig<F>

The Dropout Operation Config representation for this Plugin.

Required Methods

fn init_nn()

Initializes the Plugin.

Implementations on Foreign Types

impl<T> NN<T> for Backend<Native> where
    T: Add<T, Output = T> + Mul<T, Output = T> + Default + Copy, 

type CC = ConvolutionConfig

type CLRN = NormalizationConfig

type CPOOL = PoolingConfig

type CDROP = DropoutConfig

fn init_nn()

impl<T> NN<T> for Backend<Cuda> where
    T: Float + DataTypeInfo, 

type CC = ConvolutionConfig

type CLRN = NormalizationConfig

type CPOOL = PoolingConfig

type CDROP = DropoutConfig

fn init_nn()

Implementors