1 files changed, 8 insertions, 5 deletions
diff --git a/runtimes/pure_arm_compute/src/internal/op/Split.cc b/runtimes/pure_arm_compute/src/internal/op/Split.cc
index cbd863fce..6457a106a 100644
--- a/runtimes/pure_arm_compute/src/internal/op/Split.cc
+++ b/runtimes/pure_arm_compute/src/internal/op/Split.cc
@@ -47,13 +47,16 @@ namespace Split
 Param::Param(uint32_t inputCount, const uint32_t *inputs, uint32_t outputCount,
              const uint32_t *outputs)
 {
-  assert(inputCount == 2);
+  assert(inputCount == 3);
 
   // Each input should be interpreted as follows:
-  //  0 -> A 0-D int32 tensor, indicating the dimension along which to split.
-  //  1 -> An n-D tensor, specifying the tensor to be split.
-  axis_index = inputs[0];
-  ifm_index = inputs[1];
+  //  0 -> An n-D tensor, specifying the tensor to be split.
+  //  1 -> A 0-D int32 tensor, indicating the dimension along which to split.
+  //  2 -> A 0-D int32 tensor, indicating the number of outputs
+  //       (It can be ignored on pacl becasue pacl don't support dynamic tensor shape,
+  //        and can be used for verification only)
+  ifm_index = inputs[0];
+  axis_index = inputs[1];
 
   // Each output should be interpreted as follow:
   //  [0, outputCount) -> An n-D tensor.