aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/core/WindowIterator.h
diff options
context:
space:
mode:
Diffstat (limited to 'arm_compute/core/WindowIterator.h')
-rw-r--r--arm_compute/core/WindowIterator.h18
1 files changed, 18 insertions, 0 deletions
diff --git a/arm_compute/core/WindowIterator.h b/arm_compute/core/WindowIterator.h
index 8e58d0ce1c..32d6293a5a 100644
--- a/arm_compute/core/WindowIterator.h
+++ b/arm_compute/core/WindowIterator.h
@@ -110,6 +110,24 @@ public:
return get_ptr(x, y, z, w);
}
+ /** Returns a pointer to the first element of the tensor
+ *
+ * @return Pointer to the first element.
+ */
+ inline T *first_element()
+ {
+ return reinterpret_cast<T *>(_first);
+ }
+
+ /** Returns a pointer to the first element of the tensor
+ *
+ * @return Pointer to the first element.
+ */
+ inline T *operator()()
+ {
+ return first_element();
+ }
+
private:
uint8_t *_first; /**< Pointer to the first element of the tensor.*/
Strides _strides; /**< Strides in bytes of the tensor */