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 <bsgcomp@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
diff --git a/utils/Utils.h b/utils/Utils.h
index 8822fe7..cefb3ae 100644
--- a/utils/Utils.h
+++ b/utils/Utils.h
@@ -857,11 +857,8 @@
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);