|
void | feed (T_e *layer_mem_ptr, T_e *in_mem_ptr, T_e *out_mem_ptr) |
| apply the multi-layer perceptron on data in in_mem_ptr More...
|
|
|
static constexpr int | deduce_num_params_in () |
|
static constexpr int | deduce_num_params_hidden () |
|
static constexpr int | deduce_num_params_out () |
|
static constexpr int | deduce_num_params () |
|
◆ feed()
template<typename T_e , int T_layers_num, int T_in_num, int T_in_dim, int T_out_dim, int T_hidden_dim, opt T_opt_bias = opt::bias>
void rocky::etna::mlp< T_e, T_layers_num, T_in_num, T_in_dim, T_out_dim, T_hidden_dim, T_opt_bias >::feed |
( |
T_e * |
layer_mem_ptr, |
|
|
T_e * |
in_mem_ptr, |
|
|
T_e * |
out_mem_ptr |
|
) |
| |
|
inline |
apply the multi-layer perceptron on data in in_mem_ptr
- Parameters
-
layer_mem_ptr | memory block containing layer parameters |
in_mem_ptr | memory block containing input data |
out_mem_ptr | memory block for storing the result |
- Returns
- ** void
The documentation for this class was generated from the following file: