Home
last modified time | relevance | path

Searched refs:nn_config (Results 1 – 8 of 8) sorted by relevance

/external/libaom/libaom/av1/encoder/
Dml.c31 const NN_CONFIG *const nn_config, int reduce_prec, in av1_nn_predict_c() argument
33 int num_input_nodes = nn_config->num_inputs; in av1_nn_predict_c()
38 const int num_layers = nn_config->num_hidden_layers; in av1_nn_predict_c()
41 const float *layer_weights = nn_config->weights[layer]; in av1_nn_predict_c()
42 const float *layer_bias = nn_config->bias[layer]; in av1_nn_predict_c()
44 const int num_output_nodes = nn_config->num_hidden_nodes[layer]; in av1_nn_predict_c()
60 const float *layer_weights = nn_config->weights[num_layers]; in av1_nn_predict_c()
61 const float *layer_bias = nn_config->bias[num_layers]; in av1_nn_predict_c()
62 for (int node = 0; node < nn_config->num_outputs; ++node) { in av1_nn_predict_c()
68 if (reduce_prec) av1_nn_output_prec_reduce(output, nn_config->num_outputs); in av1_nn_predict_c()
[all …]
Dpartition_strategy.c270 const NN_CONFIG *nn_config = in av1_simple_motion_search_based_split() local
289 av1_nn_predict(features, nn_config, 1, &score); in av1_simple_motion_search_based_split()
512 const NN_CONFIG *nn_config = in av1_simple_motion_search_prune_rect() local
522 if (!nn_config || prune_thresh == 0.0f) { in av1_simple_motion_search_prune_rect()
542 av1_nn_predict(features, nn_config, 1, scores); in av1_simple_motion_search_prune_rect()
721 const NN_CONFIG *nn_config = &av1_max_part_pred_nn_config; in av1_predict_max_partition() local
727 av1_nn_predict(features, nn_config, 1, scores); in av1_predict_max_partition()
829 const NN_CONFIG *nn_config = NULL; in av1_ml_early_term_after_split() local
834 nn_config = &av1_early_term_after_split_nnconfig_64; in av1_ml_early_term_after_split()
838 nn_config = &av1_early_term_after_split_nnconfig_32; in av1_ml_early_term_after_split()
[all …]
Dml.h65 void av1_nn_predict_v2(const float *features, NN_CONFIG_V2 *nn_config,
Dtx_search.c1888 const NN_CONFIG *nn_config = av1_tx_split_nnconfig_map[tx_size]; in ml_predict_tx_split() local
1889 if (!nn_config) return -1; in ml_predict_tx_split()
1902 av1_nn_predict(features, nn_config, 1, &score); in ml_predict_tx_split()
Drdopt.c4836 const NN_CONFIG *nn_config = (AOMMIN(cm->width, cm->height) <= 480) in av1_rd_pick_inter_mode_sb() local
4852 av1_nn_predict(nn_features, nn_config, 1, scores); in av1_rd_pick_inter_mode_sb()
/external/libaom/libaom/test/
Dav1_nn_predict_test.cc29 const NN_CONFIG *const nn_config,
80 NN_CONFIG nn_config; in RunNnPredictTest() local
81 memcpy(&nn_config, shape, sizeof(nn_config)); in RunNnPredictTest()
93 nn_config.weights[i] = weights[i]; in RunNnPredictTest()
94 nn_config.bias[i] = bias[i]; in RunNnPredictTest()
120 av1_nn_predict_c(inputs, &nn_config, 0, outputs_ref); in RunNnPredictTest()
121 target_func_(inputs, &nn_config, 0, outputs_test); in RunNnPredictTest()
147 NN_CONFIG nn_config; in RunNnPredictSpeedTest() local
148 memcpy(&nn_config, shape, sizeof(nn_config)); in RunNnPredictSpeedTest()
151 nn_config.weights[i] = weights[i]; in RunNnPredictSpeedTest()
[all …]
/external/libaom/libaom/av1/encoder/x86/
Dml_sse3.c154 const NN_CONFIG *const nn_config, int reduce_prec, in av1_nn_predict_sse3() argument
158 int num_inputs = nn_config->num_inputs; in av1_nn_predict_sse3()
161 for (int layer = 0; layer <= nn_config->num_hidden_layers; layer++) { in av1_nn_predict_sse3()
162 const float *layer_weights = nn_config->weights[layer]; in av1_nn_predict_sse3()
163 const float *layer_bias = nn_config->bias[layer]; in av1_nn_predict_sse3()
164 bool output_layer = (layer == nn_config->num_hidden_layers); in av1_nn_predict_sse3()
166 const int num_outputs = output_layer ? nn_config->num_outputs in av1_nn_predict_sse3()
167 : nn_config->num_hidden_nodes[layer]; in av1_nn_predict_sse3()
243 if (reduce_prec) av1_nn_output_prec_reduce(output, nn_config->num_outputs); in av1_nn_predict_sse3()
/external/libvpx/libvpx/vp9/encoder/
Dvp9_encodeframe.c3162 static void nn_predict(const float *features, const NN_CONFIG *nn_config, in nn_predict() argument
3164 int num_input_nodes = nn_config->num_inputs; in nn_predict()
3170 const int num_layers = nn_config->num_hidden_layers; in nn_predict()
3174 const float *weights = nn_config->weights[layer]; in nn_predict()
3175 const float *bias = nn_config->bias[layer]; in nn_predict()
3177 const int num_output_nodes = nn_config->num_hidden_nodes[layer]; in nn_predict()
3195 const float *weights = nn_config->weights[num_layers]; in nn_predict()
3196 for (node = 0; node < nn_config->num_outputs; ++node) { in nn_predict()
3197 const float *bias = nn_config->bias[num_layers]; in nn_predict()
3225 const NN_CONFIG *nn_config = NULL; in ml_pruning_partition() local
[all …]