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>;
    type CRNN: NNOperationConfig<F> + RnnConfig<F>;

    // Required method
    fn init_nn();
}

Expand description

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

Required 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.

type CRNN: NNOperationConfig<F> + RnnConfig<F>

The RNN Operation Config representation for this Plugin

Required Methods§

fn init_nn()

Initializes the Plugin.

Object Safety§

This trait is not object safe.

Implementations on Foreign Types§

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

type CC = ConvolutionConfig

type CLRN = NormalizationConfig

type CPOOL = PoolingConfig

type CDROP = DropoutConfig

type CRNN = RnnConfig

fn init_nn()

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

type CRNN = RnnConfig

fn init_nn()

Implementors§