aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--utils/Utils.h5
1 files changed, 1 insertions, 4 deletions
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);