Update assertions for BatchToSpaceND op in compile time. (#3286)

This patch updates the assertions required for BatchToSpaceND at compile time in PACL.

Signed-off-by: prasannar <prasanna.r@samsung.com>

diff --git a/runtimes/pure_arm_compute/src/compilation.cc b/runtimes/pure_arm_compute/src/compilation.cc
index 1957d98..e28d879 100644 (file)
--- a/runtimes/pure_arm_compute/src/compilation.cc
+++ b/runtimes/pure_arm_compute/src/compilation.cc
@@ -3925,6 +3925,7 @@ void Planner::visit(const ::internal::tflite::op::BatchToSpaceNd::Node &node)
 
   assert(_ctx.at(input_index).shape().rank() == 4);
   assert(_ctx.at(output_index).shape().rank() == 4);
+  assert(_ctx.at(block_size_index).hasData() == true);
 
   const int32_t *block_size =
       reinterpret_cast<const int32_t *>(_ctx.at(block_size_index).data().base());
@@ -3932,7 +3933,8 @@ void Planner::visit(const ::internal::tflite::op::BatchToSpaceNd::Node &node)
   const auto &output_shape = _ctx.at(output_index).shape();
   const auto &input_shape = _ctx.at(input_index).shape();
 
-  assert(block_size[0] > 0 && block_size[1] > 0);
+  assert((_ctx.at(block_size_index).data().size() / sizeof(int32_t)) == 2 && block_size[0] > 0 &&
+         block_size[1] > 0);
   {
     assert(output_shape.dim(3) == input_shape.dim(3));
     assert(output_shape.dim(1) == input_shape.dim(1) * block_size[0]);