aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/core/Window.h
diff options
context:
space:
mode:
Diffstat (limited to 'arm_compute/core/Window.h')
-rw-r--r--arm_compute/core/Window.h21
1 files changed, 20 insertions, 1 deletions
diff --git a/arm_compute/core/Window.h b/arm_compute/core/Window.h
index 654f5ed4f8..c890bf8f9e 100644
--- a/arm_compute/core/Window.h
+++ b/arm_compute/core/Window.h
@@ -1,5 +1,5 @@
/*
- * Copyright (c) 2016, 2017 ARM Limited.
+ * Copyright (c) 2016-2018 ARM Limited.
*
* SPDX-License-Identifier: MIT
*
@@ -245,6 +245,14 @@ public:
{
return first_slice_window<3>();
};
+ /** First 4D slice of the window
+ *
+ * @return The first slice of the window.
+ */
+ Window first_slice_window_4D() const
+ {
+ return first_slice_window<4>();
+ };
/** Slide the passed 1D window slice.
*
* If slice contains the last slice then it will remain unchanged and false will be returned.
@@ -305,6 +313,17 @@ public:
*/
Window collapse_if_possible(const Window &full_window, size_t first) const;
+ /* Collapse the dimensions higher than @p first.
+ *
+ * A dimension is collapsable if it starts from 0 and matches the corresponding dimension in the full_window
+ *
+ * @param[in] full_window Full window @p window has been created from.
+ * @param[in] first Dimensions into which the following are collapsed.
+ *
+ * @return Collapsed window if successful.
+ */
+ Window collapse(const Window &full_window, size_t first) const;
+
private:
/** First slice of the window
*