IVGCVSW-2007: Fix SQUEEZE error in vts tests

Change-Id: I5475932be49419e630627ba042ea6a70151b78e7

1 files changed, 4 insertions, 4 deletions

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