Fixed unpack node inputs (#3639)

The inputs buffer contains number of splits
Following tf_ops_compatibility

Signed-off-by: Vishal Keshav <vishal.k1@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 fedc22c..82e784f 100644 (file)
--- a/runtimes/pure_arm_compute/src/internal/op/Unpack.cc
+++ b/runtimes/pure_arm_compute/src/internal/op/Unpack.cc
@@ -49,7 +49,7 @@ namespace Unpack
 Param::Param(uint32_t inputCount, const uint32_t *inputs, uint32_t outputCount,
              const uint32_t *outputs)
 {
-  assert(inputCount == 2);
+  assert(inputCount == 3);
 
   ifm_index = inputs[0];
 
@@ -58,6 +58,7 @@ 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];
 }
 
 } // namespace Unpack

diff --git a/runtimes/pure_arm_compute/src/internal/op/Unpack.h b/runtimes/pure_arm_compute/src/internal/op/Unpack.h
index 85592e3..575e3d0 100644 (file)
--- a/runtimes/pure_arm_compute/src/internal/op/Unpack.h
+++ b/runtimes/pure_arm_compute/src/internal/op/Unpack.h
@@ -35,6 +35,7 @@ struct Param
 {
   int32_t ifm_index;
   int32_t axis_index;
+  int32_t num_split_index;
   // There are N outputs after Unpacking Input Tensor along axis
   std::vector<int32_t> ofm_indexes;