From 5e69bb44adce2da6ce2db0cfb838255e1eb36aef Mon Sep 17 00:00:00 2001 From: Michalis Spyrou Date: Fri, 9 Mar 2018 16:36:00 +0000 Subject: COMPMID-959 fill_random_tensor should fill all dimensions Change-Id: I02ef28c2370e8ea525933c8c54e66ae1d764fe02 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/124017 Tested-by: Jenkins Reviewed-by: Anthony Barbier --- utils/Utils.h | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) (limited to 'utils/Utils.h') diff --git a/utils/Utils.h b/utils/Utils.h index 8822fe7121..cefb3ae4c0 100644 --- a/utils/Utils.h +++ b/utils/Utils.h @@ -857,11 +857,8 @@ void fill_random_tensor(T &tensor, float lower_bound, float upper_bound) std::random_device rd; std::mt19937 gen(rd()); - TensorShape shape(tensor.info()->dimension(0), tensor.info()->dimension(1)); - Window window; - window.set(Window::DimX, Window::Dimension(0, shape.x(), 1)); - window.set(Window::DimY, Window::Dimension(0, shape.y(), 1)); + window.use_tensor_dimensions(tensor.info()->tensor_shape()); map(tensor, true); -- cgit v1.2.1