| /external/libaom/libaom/test/ |
| D | cnn_test.cc | 36 const float *expected, const CNN_CONFIG *cnn_config, in RunCNNTest() argument
40 av1_find_cnn_output_size(image_width, image_height, cnn_config, &out_width, in RunCNNTest()
58 RunMultiOutCNNTest(&input, image_width, image_height, in_stride, cnn_config, in RunCNNTest()
66 const CNN_CONFIG *cnn_config, in RunMultiOutCNNTest() argument
77 av1_find_cnn_output_size(image_width, image_height, cnn_config, out_widths, in RunMultiOutCNNTest()
79 av1_cnn_predict(input, image_width, image_height, in_stride, cnn_config, in RunMultiOutCNNTest()
114 static void AssignLayerWeightsBiases(CNN_CONFIG *cnn_config, float *weights, in AssignLayerWeightsBiases() argument
118 for (int layer = 0; layer < cnn_config->num_layers; ++layer) { in AssignLayerWeightsBiases()
119 CNN_LAYER_CONFIG *layer_config = &cnn_config->layer_config[layer]; in AssignLayerWeightsBiases()
241 CNN_CONFIG cnn_config = { 3, in TEST_F() local
[all …]
|
| /external/libaom/libaom/av1/encoder/ |
| D | cnn.c | 221 static INLINE int cnn_has_at_least_one_output(const CNN_CONFIG *cnn_config) { in cnn_has_at_least_one_output() argument
222 const int num_layers = cnn_config->num_layers; in cnn_has_at_least_one_output()
223 const CNN_LAYER_CONFIG *layer_configs = cnn_config->layer_config; in cnn_has_at_least_one_output()
235 const CNN_CONFIG *cnn_config, int *out_width, in av1_find_cnn_output_size() argument
240 i_width[0] = in_width + cnn_config->ext_width * 2; in av1_find_cnn_output_size()
241 i_height[0] = in_height + cnn_config->ext_height * 2; in av1_find_cnn_output_size()
244 assert(cnn_has_at_least_one_output(cnn_config)); in av1_find_cnn_output_size()
247 for (int i = 0; i < cnn_config->num_layers; ++i) { in av1_find_cnn_output_size()
248 const CNN_LAYER_CONFIG *layer_config = &cnn_config->layer_config[i]; in av1_find_cnn_output_size()
833 int in_stride, const CNN_CONFIG *cnn_config, in av1_cnn_predict_c() argument
[all …]
|
| D | cnn.h | 167 const CNN_CONFIG *cnn_config, int *out_width,
173 int stride, const CNN_CONFIG *cnn_config,
178 const CNN_CONFIG *cnn_config,
185 const CNN_CONFIG *cnn_config,
189 int stride, const CNN_CONFIG *cnn_config,
|
| D | partition_strategy.c | 74 const CNN_CONFIG *cnn_config = &av1_intra_mode_cnn_partition_cnn_config; in av1_intra_mode_cnn_partition() local
121 cnn_config, &thread_data, bit_depth, in av1_intra_mode_cnn_partition()
126 av1_cnn_predict_img_multi_out(image, width, height, stride, cnn_config, in av1_intra_mode_cnn_partition()
|
| /external/libaom/libaom/av1/common/ |
| D | av1_rtcd_defs.pl | 359 …**input, int in_width, int in_height, int in_stride, const CNN_CONFIG *cnn_config, const CNN_THREA…
|
| /external/libaom/config/arm/config/ |
| D | av1_rtcd.h | 111 …**input, int in_width, int in_height, int in_stride, const CNN_CONFIG *cnn_config, const CNN_THREA…
|
| /external/libaom/config/arm64/config/ |
| D | av1_rtcd.h | 111 …**input, int in_width, int in_height, int in_stride, const CNN_CONFIG *cnn_config, const CNN_THREA…
|
| /external/libaom/config/x86/config/ |
| D | av1_rtcd.h | 111 …**input, int in_width, int in_height, int in_stride, const CNN_CONFIG *cnn_config, const CNN_THREA…
|
| /external/libaom/config/x86_64/config/ |
| D | av1_rtcd.h | 111 …**input, int in_width, int in_height, int in_stride, const CNN_CONFIG *cnn_config, const CNN_THREA…
|