Trait collenchyma_nn::NN [−] [src]

pub trait NN<F> {
    type CC: NNOperationConfig<F> + ConvolutionConfig<F>;
    type CLRN: NNOperationConfig<F>;
    type CPOOL: NNOperationConfig<F>;
    fn init_nn();
    fn device(&self) -> &DeviceType;
}

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()

Initializes the Plugin.

fn device(&self) -> &DeviceType

Returns the device on which the Plugin operations will run.

Implementors