diff options
Diffstat (limited to 'arm_compute/core/utils/misc/ShapeCalculator.h')
| -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) { |