From: 이춘석/동작제어Lab(SR)/Staff Engineer/삼성전자 <chunseok.lee@samsung.com>
Date: Thu, 9 Aug 2018 04:45:52 +0000 (+0900)
Subject: Enable squeeze for rank-4 input (#2234)
X-Git-Tag: 0.2~309
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=6b5fd65d842e6319a540d9b48e8349ed05c80c72;p=platform%2Fcore%2Fml%2Fnnfw.git

Enable squeeze for rank-4 input (#2234)

- Remove input rank restriction from compilation

Signed-off-by: Chunseok Lee <chunseok.lee@samsung.com>
---

diff --git a/runtimes/pure_arm_compute/src/compilation.cc b/runtimes/pure_arm_compute/src/compilation.cc
index 1481096..c07527c 100644
--- a/runtimes/pure_arm_compute/src/compilation.cc
+++ b/runtimes/pure_arm_compute/src/compilation.cc
@@ -2008,7 +2008,6 @@ void Planner::visit(const ::internal::tflite::op::Squeeze::Node &node)
   //   - 3D input : squeeze_float_1
   //   - 2D input : squeeze_3D_float_1
   //   - 4D input fails (squeeze.mod.py) -> we need general tensor support
-  assert(_ctx.at(input_index).shape().rank() < 4);
 
   // TODO Support generic tensor shape
   const auto output_shape = _ctx.at(output_index).shape();