Trait coaster_nn::NN [−][src]
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>;
fn init_nn();
}
Expand description
Provides the functionality for a backend to support Neural Network related operations.
Associated Types
type CC: NNOperationConfig<F> + ConvolutionConfig<F>
type CC: NNOperationConfig<F> + ConvolutionConfig<F>
The Convolution Operation Config representation for this Plugin.
type CLRN: NNOperationConfig<F>
type CLRN: NNOperationConfig<F>
The LRN Operation Config representation for this Plugin.
type CPOOL: NNOperationConfig<F>
type CPOOL: NNOperationConfig<F>
The Pooling Operation Config representation for this Plugin.
type CDROP: NNOperationConfig<F>
type CDROP: NNOperationConfig<F>
The Dropout Operation Config representation for this Plugin.
type CRNN: NNOperationConfig<F> + RnnConfig<F>
type CRNN: NNOperationConfig<F> + RnnConfig<F>
The RNN Operation Config representation for this Plugin