diff options
author | Pablo Tello <pablo.tello@arm.com> | 2017-07-05 15:20:38 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-09-17 14:16:42 +0100 |
commit | 8fda1cb6f4142133fff045a6f9c18778757c316c (patch) | |
tree | 3f0ad562b24cc3c76e8a745cb59cd584b664ec57 /tests/NEON/Helper.h | |
parent | 8df3fafde3dcf131def3471db8e8b1a1c34b354b (diff) | |
download | ComputeLibrary-8fda1cb6f4142133fff045a6f9c18778757c316c.tar.gz |
COMPMID-421: Added FP16 support in BatchNormalizationLayer.
Change-Id: I7142e0e8466ef79e016ae56d285e8e9291573e52
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/79814
Reviewed-by: Moritz Pflanzer <moritz.pflanzer@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Diffstat (limited to 'tests/NEON/Helper.h')
-rw-r--r-- | tests/NEON/Helper.h | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/tests/NEON/Helper.h b/tests/NEON/Helper.h index 5b0f750fca..9e60a48ecd 100644 --- a/tests/NEON/Helper.h +++ b/tests/NEON/Helper.h @@ -25,8 +25,10 @@ #define __ARM_COMPUTE_TEST_NEON_HELPER_H__ #include "arm_compute/runtime/Array.h" +#include "tests/Globals.h" #include <algorithm> +#include <array> #include <vector> namespace arm_compute @@ -44,6 +46,20 @@ Array<T> create_array(const std::vector<T> &v) return array; } + +template <typename D, typename T, typename... Ts> +void fill_tensors(D &&dist, std::initializer_list<int> seeds, T &&tensor, Ts &&... other_tensors) +{ + const std::array < T, 1 + sizeof...(Ts) > tensors{ { std::forward<T>(tensor), std::forward<Ts>(other_tensors)... } }; + std::vector<int> vs(seeds); + ARM_COMPUTE_ERROR_ON(vs.size() != tensors.size()); + int k = 0; + for(auto tp : tensors) + { + library->fill(Accessor(*tp), std::forward<D>(dist), vs[k++]); + } +} + } // namespace test } // namespace arm_compute #endif /* __ARM_COMPUTE_TEST_NEON_HELPER_H__ */ |