diff options
author | Guo, Yejun <yejun.guo@intel.com> | 2019-08-29 13:53:33 +0800 |
---|---|---|
committer | Pedro Arthur <bygrandao@gmail.com> | 2019-08-30 11:41:30 -0300 |
commit | 09a455a24649af36e8eea81029be7a410201be4c (patch) | |
tree | 70d0e8e1575068d010bc229864cf5badb5595fb8 /libavfilter/dnn/dnn_backend_native_layer_pad.h | |
parent | 20a12448aaf140cf1ec06ee22e1ab4ebaea5c9ba (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.h | 4 |
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 |