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>
diff --git a/arm_compute/core/utils/misc/ShapeCalculator.h b/arm_compute/core/utils/misc/ShapeCalculator.h
index adf5309..4756ff4 100644
--- a/arm_compute/core/utils/misc/ShapeCalculator.h
+++ b/arm_compute/core/utils/misc/ShapeCalculator.h
@@ -803,6 +803,13 @@
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)
{