From a893efa3494121f910058ae5b75e32bf6c1d5720 Mon Sep 17 00:00:00 2001
From: saoste01 <saoirse.stewart@arm.com>
Date: Sat, 13 Oct 2018 11:56:12 +0100
Subject: IVGCVSW-2007: Fix SQUEEZE error in vts tests

Change-Id: I5475932be49419e630627ba042ea6a70151b78e7
---
 1.1/HalPolicy.cpp | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

(limited to '1.1')

diff --git a/1.1/HalPolicy.cpp b/1.1/HalPolicy.cpp
index 1b1c06ea..d241290f 100644
--- a/1.1/HalPolicy.cpp
+++ b/1.1/HalPolicy.cpp
@@ -300,19 +300,19 @@ bool HalPolicy::ConvertSqueeze(const Operation& operation, const Model& model, C
     if(!axisOperand)
     {
         axis.assign(dimensionSequence,
-                    dimensionSequence+inputInfo.GetNumDimensions());
+                    dimensionSequence+rank);
     }
     else
     {
         GetTensorInt32Values(*axisOperand, axis, model, data);
     }
 
+
     std::vector<uint32_t> outputDims;
-    for (auto& i : axis)
+    for(unsigned int i = 0; i < rank; i++)
     {
-        auto currentDimension = inputInfo.GetShape()[i];
         bool skipSqueeze = (std::find(axis.begin(), axis.end(), i) == axis.end());
-
+        auto currentDimension = inputInfo.GetShape()[i];
         if (skipSqueeze || currentDimension != 1)
         {
             outputDims.push_back(currentDimension);
-- 
cgit v1.2.1