From 60cfea53a05b4718b0c49ce560c794d07a8327de Mon Sep 17 00:00:00 2001 From: chxin66 <57057788+chxin66@users.noreply.github.com> Date: Fri, 6 May 2022 09:31:14 +0800 Subject: [PATCH] fix gather_element operation input num issue (#388) Change-Id: Id2e685cf6993776e6674f528b71eb842420b16ad Author: Xia Kaihong Date: Thu Apr 14 16:23:16 2022 +0800 --- src/tim/vx/internal/src/ops/vsi_nn_op_gather_elements.c | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/tim/vx/internal/src/ops/vsi_nn_op_gather_elements.c b/src/tim/vx/internal/src/ops/vsi_nn_op_gather_elements.c index 2ba3c5c..baf55b1 100644 --- a/src/tim/vx/internal/src/ops/vsi_nn_op_gather_elements.c +++ b/src/tim/vx/internal/src/ops/vsi_nn_op_gather_elements.c @@ -46,7 +46,7 @@ typedef struct _gather_elements_local_data_t { /* Declare number of input and output. */ -#define _INPUT_NUM (1) +#define _INPUT_NUM (2) #define _OUTPUT_NUM (1) static vsi_status op_compute