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

The Convolution Operation Config representation for this Plugin.

The LRN Operation Config representation for this Plugin.

The Pooling Operation Config representation for this Plugin.

Required Methods

Initializes the Plugin.

Returns the device on which the Plugin operations will run.

Implementors