diff options
author | Pablo Tello <pablo.tello@arm.com> | 2019-03-05 17:33:08 +0000 |
---|---|---|
committer | Pablo Marquez <pablo.tello@arm.com> | 2019-03-20 13:29:51 +0000 |
commit | 6a14adb2faefd2f4638647595dfdef12e29f8a01 (patch) | |
tree | 56b88d858f0521ace2dc44b49b42270d996b9bad /arm_compute/core/utils/misc | |
parent | 341b218eff8c75aa9862b333bb0012a03c14a1ef (diff) | |
download | ComputeLibrary-6a14adb2faefd2f4638647595dfdef12e29f8a01.tar.gz |
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>
Diffstat (limited to 'arm_compute/core/utils/misc')
-rw-r--r-- | arm_compute/core/utils/misc/ShapeCalculator.h | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/arm_compute/core/utils/misc/ShapeCalculator.h b/arm_compute/core/utils/misc/ShapeCalculator.h index 6782cda7fe..7a34b43028 100644 --- a/arm_compute/core/utils/misc/ShapeCalculator.h +++ b/arm_compute/core/utils/misc/ShapeCalculator.h @@ -1239,6 +1239,7 @@ inline TensorShape calculate_concatenate_shape(const std::vector<T *> &input, si { ARM_COMPUTE_ERROR_ON(tensor == nullptr); const TensorShape shape = extract_shape(tensor); + ARM_COMPUTE_ERROR_ON(axis >= shape.num_dimensions()); new_size += shape[axis]; } |