diff options
Diffstat (limited to 'media/libaom/src/av1/encoder/cnn.h')
| -rw-r--r-- | media/libaom/src/av1/encoder/cnn.h | 14 |
1 files changed, 10 insertions, 4 deletions
diff --git a/media/libaom/src/av1/encoder/cnn.h b/media/libaom/src/av1/encoder/cnn.h index 706be44474..1a6c03a4c9 100644 --- a/media/libaom/src/av1/encoder/cnn.h +++ b/media/libaom/src/av1/encoder/cnn.h @@ -17,6 +17,7 @@ extern "C" { #endif #include <math.h> +#include <stdbool.h> #include "aom_util/aom_thread.h" #include "config/av1_rtcd.h" @@ -167,13 +168,18 @@ void av1_find_cnn_output_size(int in_width, int in_height, const CNN_CONFIG *cnn_config, int *out_width, int *out_height, int *out_channels); +// Function to return output width and output height of given layer. +void av1_find_cnn_layer_output_size(int in_width, int in_height, + const CNN_LAYER_CONFIG *layer_config, + int *out_width, int *out_height); + // Prediction functions from set of input image buffers. This function supports // CNN with multiple outputs. -void av1_cnn_predict_img_multi_out(uint8_t **dgd, int width, int height, +bool av1_cnn_predict_img_multi_out(uint8_t **dgd, int width, int height, int stride, const CNN_CONFIG *cnn_config, const CNN_THREAD_DATA *thread_data, struct CNN_MULTI_OUT *output); -void av1_cnn_predict_img_multi_out_highbd(uint16_t **dgd, int width, int height, +bool av1_cnn_predict_img_multi_out_highbd(uint16_t **dgd, int width, int height, int stride, const CNN_CONFIG *cnn_config, const CNN_THREAD_DATA *thread_data, @@ -181,11 +187,11 @@ void av1_cnn_predict_img_multi_out_highbd(uint16_t **dgd, int width, int height, // Prediction functions from set of input image buffers. This function only // supports a single output. -void av1_cnn_predict_img(uint8_t **dgd, int width, int height, int stride, +bool av1_cnn_predict_img(uint8_t **dgd, int width, int height, int stride, const CNN_CONFIG *cnn_config, const CNN_THREAD_DATA *thread_data, float **output, int out_stride); -void av1_cnn_predict_img_highbd(uint16_t **dgd, int width, int height, +bool av1_cnn_predict_img_highbd(uint16_t **dgd, int width, int height, int stride, const CNN_CONFIG *cnn_config, const CNN_THREAD_DATA *thread_data, int bit_depth, float **output, int out_stride); |