Searched refs:NN_MAX_NODES_PER_LAYER (Results 1 – 6 of 6) sorted by relevance
37 const int MAX_NODES2 = NN_MAX_NODES_PER_LAYER * NN_MAX_NODES_PER_LAYER; in SetUp()43 (float *)aom_malloc(NN_MAX_NODES_PER_LAYER * in SetUp()49 bias[i] = &bias_buf[i * NN_MAX_NODES_PER_LAYER]; in SetUp()74 float inputs[NN_MAX_NODES_PER_LAYER] = { 0 }; in RunNnPredictTest()75 float outputs_test[NN_MAX_NODES_PER_LAYER] = { 0 }; in RunNnPredictTest()76 float outputs_ref[NN_MAX_NODES_PER_LAYER] = { 0 }; in RunNnPredictTest()100 for (int node = 0; node < NN_MAX_NODES_PER_LAYER; node++) { in RunNnPredictTest()103 for (int node = 0; node < NN_MAX_NODES_PER_LAYER * NN_MAX_NODES_PER_LAYER; in RunNnPredictTest()110 for (int node = 0; node < NN_MAX_NODES_PER_LAYER; node++) { in RunNnPredictTest()113 for (int node = 0; node < NN_MAX_NODES_PER_LAYER * NN_MAX_NODES_PER_LAYER; in RunNnPredictTest()[all …]
25 float buf[2][NN_MAX_NODES_PER_LAYER]; in av1_nn_predict_c()35 assert(num_output_nodes < NN_MAX_NODES_PER_LAYER); in av1_nn_predict_c()
22 #define NN_MAX_NODES_PER_LAYER 128 macro
19 #define NN_MAX_NODES_PER_LAYER 128 macro
3079 float buf[2][NN_MAX_NODES_PER_LAYER]; in nn_predict()3091 assert(num_output_nodes < NN_MAX_NODES_PER_LAYER); in nn_predict()
156 float buf[2][NN_MAX_NODES_PER_LAYER]; in av1_nn_predict_sse3()