summaryrefslogtreecommitdiff
path: root/chromium/third_party/ffmpeg/libavfilter/dnn_backend_native.h
diff options
context:
space:
mode:
Diffstat (limited to 'chromium/third_party/ffmpeg/libavfilter/dnn_backend_native.h')
-rw-r--r--chromium/third_party/ffmpeg/libavfilter/dnn_backend_native.h8
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);