COMPMID-443: Collapse higher dimensions for activation layer
Change-Id: I5943235aff1bb6440e3ab08e818d53aa5d94143a
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/80349
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
diff --git a/arm_compute/core/Window.h b/arm_compute/core/Window.h
index 6e7ef22..a2fbc8f 100644
--- a/arm_compute/core/Window.h
+++ b/arm_compute/core/Window.h
@@ -326,6 +326,17 @@
return _num_threads;
}
+ /* Collapse the dimensions higher than @p first if possible.
+ *
+ * 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.
+ */
+ Window collapse_if_possible(const Window &full_window, size_t first) const;
+
private:
/** First slice of the window
*