commit | 6a14adb2faefd2f4638647595dfdef12e29f8a01 | [log] [tgz] |
---|---|---|
author | Pablo Tello <pablo.tello@arm.com> | Tue Mar 05 17:33:08 2019 +0000 |
committer | Pablo Marquez <pablo.tello@arm.com> | Wed Mar 20 13:29:51 2019 +0000 |
tree | 56b88d858f0521ace2dc44b49b42270d996b9bad | |
parent | 341b218eff8c75aa9862b333bb0012a03c14a1ef [diff] [blame] |
COMPMID-1932: Implement CLHeightConcatenateLayer. Change-Id: I244ff2352c3c9620148dab6475aee62b218b96ef Signed-off-by: Pablo Tello <pablo.tello@arm.com> Reviewed-on: https://review.mlplatform.org/c/871 Tested-by: Arm Jenkins <bsgcomp@arm.com> Comments-Addressed: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
diff --git a/arm_compute/core/utils/misc/ShapeCalculator.h b/arm_compute/core/utils/misc/ShapeCalculator.h index 6782cda..7a34b43 100644 --- a/arm_compute/core/utils/misc/ShapeCalculator.h +++ b/arm_compute/core/utils/misc/ShapeCalculator.h
@@ -1239,6 +1239,7 @@ { ARM_COMPUTE_ERROR_ON(tensor == nullptr); const TensorShape shape = extract_shape(tensor); + ARM_COMPUTE_ERROR_ON(axis >= shape.num_dimensions()); new_size += shape[axis]; }