summaryrefslogtreecommitdiff
path: root/libavfilter/dnn/dnn_backend_native_layer_pad.h
diff options
context:
space:
mode:
authorGuo, Yejun <yejun.guo@intel.com>2019-08-29 13:53:33 +0800
committerPedro Arthur <bygrandao@gmail.com>2019-08-30 11:41:30 -0300
commit09a455a24649af36e8eea81029be7a410201be4c (patch)
tree70d0e8e1575068d010bc229864cf5badb5595fb8 /libavfilter/dnn/dnn_backend_native_layer_pad.h
parent20a12448aaf140cf1ec06ee22e1ab4ebaea5c9ba (diff)
dnn: introduce dnn operand (in c code) to hold operand infos within network
the info can be saved in dnn operand object without regenerating again and again, and it is also needed for layer split/merge, and for memory reuse. to make things step by step, this patch just focuses on c code, the change within python script will be added later. Signed-off-by: Guo, Yejun <yejun.guo@intel.com> Signed-off-by: Pedro Arthur <bygrandao@gmail.com>
Diffstat (limited to 'libavfilter/dnn/dnn_backend_native_layer_pad.h')
-rw-r--r--libavfilter/dnn/dnn_backend_native_layer_pad.h4
1 files changed, 3 insertions, 1 deletions
diff --git a/libavfilter/dnn/dnn_backend_native_layer_pad.h b/libavfilter/dnn/dnn_backend_native_layer_pad.h
index 0fbe652202..7cc8213521 100644
--- a/libavfilter/dnn/dnn_backend_native_layer_pad.h
+++ b/libavfilter/dnn/dnn_backend_native_layer_pad.h
@@ -26,6 +26,7 @@
#define AVFILTER_DNN_DNN_BACKEND_NATIVE_LAYER_PAD_H
#include <stdint.h>
+#include "dnn_backend_native.h"
typedef enum {LPMP_CONSTANT, LPMP_REFLECT, LPMP_SYMMETRIC} LayerPadModeParam;
@@ -35,6 +36,7 @@ typedef struct LayerPadParams{
float constant_values;
} LayerPadParams;
-void dnn_execute_layer_pad(const float *input, float *output, const LayerPadParams *params, int number, int height, int width, int channel);
+int dnn_execute_layer_pad(DnnOperand *operands, const int32_t *input_operand_indexes, int32_t output_operand_index,
+ const LayerPadParams *params);
#endif