diff options
author | Guo, Yejun <yejun.guo@intel.com> | 2019-10-09 22:08:11 +0800 |
---|---|---|
committer | Pedro Arthur <bygrandao@gmail.com> | 2019-10-15 18:56:25 -0300 |
commit | 3fd5ac7e92049b4f31026acdb53a762289f71448 (patch) | |
tree | 49916c1eca3e45d745c85a10da232c0d981fb3dd /libavfilter/dnn/dnn_backend_native_layer_depth2space.c | |
parent | b78dc27bba2cc612643df7e9c84addc142273e71 (diff) | |
download | ffmpeg-streaming-3fd5ac7e92049b4f31026acdb53a762289f71448.zip ffmpeg-streaming-3fd5ac7e92049b4f31026acdb53a762289f71448.tar.gz |
avfilter/dnn: unify the layer execution function in native mode
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_depth2space.c')
-rw-r--r-- | libavfilter/dnn/dnn_backend_native_layer_depth2space.c | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/libavfilter/dnn/dnn_backend_native_layer_depth2space.c b/libavfilter/dnn/dnn_backend_native_layer_depth2space.c index a248764..3720060 100644 --- a/libavfilter/dnn/dnn_backend_native_layer_depth2space.c +++ b/libavfilter/dnn/dnn_backend_native_layer_depth2space.c @@ -27,9 +27,12 @@ #include "libavutil/avassert.h" #include "dnn_backend_native_layer_depth2space.h" -int depth_to_space(DnnOperand *operands, const int32_t *input_operand_indexes, int32_t output_operand_index, int block_size) +int dnn_execute_layer_depth2space(DnnOperand *operands, const int32_t *input_operand_indexes, + int32_t output_operand_index, const void *parameters) { float *output; + const DepthToSpaceParams *params = (const DepthToSpaceParams *)parameters; + int block_size = params->block_size; int32_t input_operand_index = input_operand_indexes[0]; int number = operands[input_operand_index].dims[0]; int height = operands[input_operand_index].dims[1]; |