Trait coaster_nn::NN

pub trait NN<F> {
    type CC: NNOperationConfig<F> + ConvolutionConfig<F>;
    type CLRN: NNOperationConfig<F>;
    type CPOOL: 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.

Required Methods

fn init_nn()

The Activation Operation Config representation for this Plugin. Initializes the Plugin.

Implementors

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