From 7cdddd13876f0525157fe49ea97396c9ea5b18eb Mon Sep 17 00:00:00 2001 From: =?utf8?q?Prasanna=20R/System=20SW=20/SRI-Bangalore/Engineer/?= =?utf8?q?=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Wed, 24 Oct 2018 16:33:02 +0530 Subject: [PATCH] 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 --- runtimes/pure_arm_compute/src/compilation.cc | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/runtimes/pure_arm_compute/src/compilation.cc b/runtimes/pure_arm_compute/src/compilation.cc index 1957d98..e28d879 100644 --- 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(_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]); -- 2.7.4