summaryrefslogtreecommitdiffstats
path: root/libavfilter/dnn/dnn_backend_native_layer_depth2space.c
diff options
context:
space:
mode:
authorGuo, Yejun <yejun.guo@intel.com>2019-10-09 22:08:11 +0800
committerPedro Arthur <bygrandao@gmail.com>2019-10-15 18:56:25 -0300
commit3fd5ac7e92049b4f31026acdb53a762289f71448 (patch)
tree49916c1eca3e45d745c85a10da232c0d981fb3dd /libavfilter/dnn/dnn_backend_native_layer_depth2space.c
parentb78dc27bba2cc612643df7e9c84addc142273e71 (diff)
downloadffmpeg-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.c5
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];
OpenPOWER on IntegriCloud