From 543036904d5324c4738cdc8b586c301471046c8e Mon Sep 17 00:00:00 2001
From: Pablo Tello <pablo.tello@arm.com>
Date: Thu, 22 Nov 2018 16:14:36 +0000
Subject: COMPMID-1726: Implement CLUnstack.

Change-Id: I94b0707d19757c5f5d7ca66d9c47e378867126a3
Reviewed-on: https://review.mlplatform.org/325
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
---
 arm_compute/core/utils/misc/ShapeCalculator.h | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'arm_compute/core/utils')

diff --git a/arm_compute/core/utils/misc/ShapeCalculator.h b/arm_compute/core/utils/misc/ShapeCalculator.h
index adf5309ea5..4756ff4f97 100644
--- a/arm_compute/core/utils/misc/ShapeCalculator.h
+++ b/arm_compute/core/utils/misc/ShapeCalculator.h
@@ -803,6 +803,13 @@ inline TensorShape extract_shape(const TensorShape *data)
     return *data;
 }
 
+inline TensorShape calculate_unstack_shape(TensorShape input_shape, unsigned int axis)
+{
+    ARM_COMPUTE_ERROR_ON(axis > input_shape.num_dimensions());
+    input_shape.remove_dimension(axis);
+    return input_shape;
+}
+
 template <typename T>
 inline TensorShape calculate_depth_concatenate_shape(const std::vector<T *> &inputs_vector)
 {
-- 
cgit v1.2.1