aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/core/Dimensions.h
diff options
context:
space:
mode:
Diffstat (limited to 'arm_compute/core/Dimensions.h')
-rw-r--r--arm_compute/core/Dimensions.h11
1 files changed, 11 insertions, 0 deletions
diff --git a/arm_compute/core/Dimensions.h b/arm_compute/core/Dimensions.h
index 3d9a3fa7ff..912b9d57d7 100644
--- a/arm_compute/core/Dimensions.h
+++ b/arm_compute/core/Dimensions.h
@@ -141,6 +141,17 @@ public:
std::fill(_id.begin() + _num_dimensions, _id.end(), 0);
}
+ /** Collapse dimensions starting from a given point
+ *
+ * @param[in] start Starting point of collapsing dimensions
+ */
+ void collapse_from(size_t start)
+ {
+ ARM_COMPUTE_ERROR_ON(start > num_dimensions());
+
+ collapse(num_dimensions() - start, start);
+ }
+
/** Returns a read/write iterator that points to the first element in the dimension array. */
typename std::array<T, num_max_dimensions>::iterator begin()
{