HMLP: High-performance Machine Learning Primitives
|
Public Member Functions | |
LayerBase (size_t user_N, size_t user_B) | |
size_t | NeuronSize () |
size_t | BatchSize () |
virtual size_t | ParameterSize ()=0 |
virtual void | Forward ()=0 |
virtual Data< T > & | GetValues ()=0 |
|
pure virtual |
The input and output types are subjected to change.
Implemented in hmlp::Layer< FC, T >, and hmlp::Layer< INPUT, T >.
|
inline |
Number of neurons