aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/core/TensorShape.h
diff options
context:
space:
mode:
authorIsabella Gottardi <isabella.gottardi@arm.com>2018-07-16 19:02:47 +0100
committerAnthony Barbier <anthony.barbier@arm.com>2018-11-02 16:54:54 +0000
commit5f29d4aecfe1cd2f685e55e4ae00da14cbb5245a (patch)
treeab88e1205b1ce766a70efa35832f3961ab7c3d3c /arm_compute/core/TensorShape.h
parent42447c120809ee6e767cdeda8a2e52d011519b1d (diff)
downloadComputeLibrary-5f29d4aecfe1cd2f685e55e4ae00da14cbb5245a.tar.gz
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>
Diffstat (limited to 'arm_compute/core/TensorShape.h')
-rw-r--r--arm_compute/core/TensorShape.h2
1 files changed, 1 insertions, 1 deletions
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;
}