Searched refs:NN_MAX_HIDDEN_LAYERS (Results 1 – 5 of 5) sorted by relevance
21 #define NN_MAX_HIDDEN_LAYERS 10 macro29 int num_hidden_nodes[NN_MAX_HIDDEN_LAYERS];31 const float *weights[NN_MAX_HIDDEN_LAYERS + 1];33 const float *bias[NN_MAX_HIDDEN_LAYERS + 1];56 FC_LAYER layer[NN_MAX_HIDDEN_LAYERS + 1]; // The layer array
39 assert(num_layers <= NN_MAX_HIDDEN_LAYERS); in av1_nn_predict_c()126 assert(num_layers <= NN_MAX_HIDDEN_LAYERS); in av1_nn_predict_v2()
42 weights_buf = (float *)aom_malloc(MAX_NODES2 * (NN_MAX_HIDDEN_LAYERS + 1) * in SetUp()46 (NN_MAX_HIDDEN_LAYERS + 1) * sizeof(*bias_buf)); in SetUp()49 for (int i = 0; i < NN_MAX_HIDDEN_LAYERS + 1; i++) { in SetUp()69 float *weights[NN_MAX_HIDDEN_LAYERS + 1] = { 0 };70 float *bias[NN_MAX_HIDDEN_LAYERS + 1] = { 0 };92 for (int i = 0; i < NN_MAX_HIDDEN_LAYERS + 1; i++) { in RunNnPredictTest()150 for (int i = 0; i < NN_MAX_HIDDEN_LAYERS; i++) { in RunNnPredictSpeedTest()
18 #define NN_MAX_HIDDEN_LAYERS 10 macro29 int num_hidden_nodes[NN_MAX_HIDDEN_LAYERS];31 const float *weights[NN_MAX_HIDDEN_LAYERS + 1];33 const float *bias[NN_MAX_HIDDEN_LAYERS + 1];
3172 assert(num_layers <= NN_MAX_HIDDEN_LAYERS); in nn_predict()