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 (file)
--- 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();