diff options
Diffstat (limited to 'chromium/third_party/ffmpeg/libavfilter/dnn_backend_native.h')
-rw-r--r-- | chromium/third_party/ffmpeg/libavfilter/dnn_backend_native.h | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/chromium/third_party/ffmpeg/libavfilter/dnn_backend_native.h b/chromium/third_party/ffmpeg/libavfilter/dnn_backend_native.h index 51d4cac9553..59179557336 100644 --- a/chromium/third_party/ffmpeg/libavfilter/dnn_backend_native.h +++ b/chromium/third_party/ffmpeg/libavfilter/dnn_backend_native.h @@ -32,7 +32,9 @@ typedef enum {INPUT, CONV, DEPTH_TO_SPACE} DNNLayerType; -typedef enum {RELU, TANH, SIGMOID} DNNActivationFunc; +typedef enum {RELU, TANH, SIGMOID, NONE, LEAKY_RELU} DNNActivationFunc; + +typedef enum {VALID, SAME, SAME_CLAMP_TO_EDGE} DNNConvPaddingParam; typedef struct Layer{ DNNLayerType type; @@ -43,6 +45,8 @@ typedef struct Layer{ typedef struct ConvolutionalParams{ int32_t input_num, output_num, kernel_size; DNNActivationFunc activation; + DNNConvPaddingParam padding_method; + int32_t dilation; float *kernel; float *biases; } ConvolutionalParams; @@ -63,7 +67,7 @@ typedef struct ConvolutionalNetwork{ DNNModel *ff_dnn_load_model_native(const char *model_filename); -DNNReturnType ff_dnn_execute_model_native(const DNNModel *model); +DNNReturnType ff_dnn_execute_model_native(const DNNModel *model, DNNData *outputs, uint32_t nb_output); void ff_dnn_free_model_native(DNNModel **model); |