From 5f29d4aecfe1cd2f685e55e4ae00da14cbb5245a Mon Sep 17 00:00:00 2001
From: Isabella Gottardi <isabella.gottardi@arm.com>
Date: Mon, 16 Jul 2018 19:02:47 +0100
Subject: COMPMID-1246 - Fix TensorShape collapse_from

Change-Id: I076f5e43c76446515d0be7bb1a5d4540d0894760
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/140168
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>
---
 arm_compute/core/TensorShape.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'arm_compute/core/TensorShape.h')

diff --git a/arm_compute/core/TensorShape.h b/arm_compute/core/TensorShape.h
index 0340e1a644..d531dcf462 100644
--- a/arm_compute/core/TensorShape.h
+++ b/arm_compute/core/TensorShape.h
@@ -160,7 +160,7 @@ public:
     TensorShape collapsed_from(size_t start) const
     {
         TensorShape copy(*this);
-        copy.collapse(num_dimensions(), start);
+        copy.collapse(num_dimensions() - start, start);
         return copy;
     }
 
-- 
cgit v1.2.1