From 671a11e1c8e1e4db7bcae9ce97b0c97ebcb97464 Mon Sep 17 00:00:00 2001 From: Anthony Barbier Date: Fri, 6 Jul 2018 15:11:36 +0100 Subject: 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 Reviewed-by: Georgios Pinitas --- arm_compute/core/Window.h | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) (limited to 'arm_compute/core/Window.h') 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:
-- cgit v1.2.1