NN

Trait NN 

Source
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§

Source

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

The Convolution Operation Config representation for this Plugin.

Source

type CLRN: NNOperationConfig<F>

The LRN Operation Config representation for this Plugin.

Source

type CPOOL: NNOperationConfig<F>

The Pooling Operation Config representation for this Plugin.

Source

type CDROP: NNOperationConfig<F>

The Dropout Operation Config representation for this Plugin.

Source

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

The RNN Operation Config representation for this Plugin

Required Methods§

Source

fn init_nn()

Initializes the Plugin.

Dyn Compatibility§

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.

Implementations on Foreign Types§

Source§

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

Source§

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

Implementors§