diff options
| author | Pablo Tello <pablo.tello@arm.com> | 2018-11-22 16:14:36 +0000 |
|---|---|---|
| committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-12-21 18:28:31 +0000 |
| commit | 543036904d5324c4738cdc8b586c301471046c8e (patch) | |
| tree | 680efe9d7d02b23a0453495c13f4ece5d0a0edbd /arm_compute/core/utils | |
| parent | 8b6b4a959a49127d64293f8b60265f0f5ed486d4 (diff) | |
| download | ComputeLibrary-543036904d5324c4738cdc8b586c301471046c8e.tar.gz | |
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>
Diffstat (limited to 'arm_compute/core/utils')
| -rw-r--r-- | arm_compute/core/utils/misc/ShapeCalculator.h | 7 |
1 files changed, 7 insertions, 0 deletions
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) { |