diff options
| author | Anthony Barbier <anthony.barbier@arm.com> | 2018-07-06 15:11:36 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:54:10 +0000 |
| commit | 671a11e1c8e1e4db7bcae9ce97b0c97ebcb97464 (patch) | |
| tree | 6bdeaf330a81e3f4f35c34817fe594d6fbc81897 /arm_compute/core/Window.h | |
| parent | 42a31723ebe79895c9bb2297a9c2ef22c01a6f26 (diff) | |
| download | ComputeLibrary-671a11e1c8e1e4db7bcae9ce97b0c97ebcb97464.tar.gz | |
COMPMID-1379: Created WindowIterator and TensorAccessor
- WindowIterator: used to iterate over arbitrary positions of a window. (More flexible than execute_window_loop which only can iterate over entire dimensions)
- TensorAccessor: RSH's code uses pointers to specialised types and strides in element sizes, this helps interfacing with their code.
Change-Id: I8ded8758d345668804873409f949b8cec694d289
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/139082
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Diffstat (limited to 'arm_compute/core/Window.h')
| -rw-r--r-- | arm_compute/core/Window.h | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/arm_compute/core/Window.h b/arm_compute/core/Window.h index 6f172ecebf..73c8d4385b 100644 --- a/arm_compute/core/Window.h +++ b/arm_compute/core/Window.h @@ -224,7 +224,13 @@ public: * @return The number of iterations */ constexpr size_t num_iterations(size_t dimension) const; - + /** Return the total number of iterations needed to iterate through the entire window + * + * @return Number of total iterations + */ + size_t num_iterations_total() const; + /** Return the shape of the window in number of steps */ + TensorShape shape() const; /** Split a window into a set of sub windows along a given dimension * * For example to split a window into 3 sub-windows along the Y axis, you would have to do:<br/> |