Correcting the order of input in unpack op (#3868)

This patch corrects the order in which the unpack op takes the input as suggested in #3787.

Signed-off-by: b.devansh <b.devansh@samsung.com>

diff --git a/runtimes/pure_arm_compute/src/internal/op/Unpack.cc b/runtimes/pure_arm_compute/src/internal/op/Unpack.cc
index 82e784f..7f4fb04 100644 (file)
--- a/runtimes/pure_arm_compute/src/internal/op/Unpack.cc
+++ b/runtimes/pure_arm_compute/src/internal/op/Unpack.cc
@@ -57,8 +57,8 @@ Param::Param(uint32_t inputCount, const uint32_t *inputs, uint32_t outputCount,
   {
     ofm_indexes.emplace_back(outputs[n]);
   }
-  axis_index = inputs[1];
-  num_split_index = inputs[2];
+  num_split_index = inputs[1];
+  axis_index = inputs[2];
 }
 
 } // namespace Unpack