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