diff options
author | Michalis Spyrou <michalis.spyrou@arm.com> | 2018-03-09 16:36:00 +0000 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:49:16 +0000 |
commit | 5e69bb44adce2da6ce2db0cfb838255e1eb36aef (patch) | |
tree | 6b07a122b219697785bdf5db88a9c37af4394d3e | |
parent | 1f9ca1d7737846c74053d68ee0844b448bae298b (diff) | |
download | ComputeLibrary-5e69bb44adce2da6ce2db0cfb838255e1eb36aef.tar.gz |
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>
-rw-r--r-- | utils/Utils.h | 5 |
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); |