diff options
Diffstat (limited to 'arm_compute/core/WindowIterator.h')
-rw-r--r-- | arm_compute/core/WindowIterator.h | 18 |
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 */ |