diff options
Diffstat (limited to 'tests/NEON/Helper.h')
-rw-r--r-- | tests/NEON/Helper.h | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/tests/NEON/Helper.h b/tests/NEON/Helper.h index c8f1c2e635..e64ac27ef3 100644 --- a/tests/NEON/Helper.h +++ b/tests/NEON/Helper.h @@ -27,6 +27,7 @@ #include "Globals.h" #include "TensorLibrary.h" +#include "arm_compute/runtime/Array.h" #include "arm_compute/runtime/Tensor.h" namespace arm_compute @@ -71,6 +72,18 @@ inline Tensor create_tensor(const std::string &name, DataType data_type, int fix return tensor; } + +template <typename T> +Array<T> create_array(const std::vector<T> &v) +{ + const size_t v_size = v.size(); + Array<T> array(v_size); + + array.resize(v_size); + std::copy(v.begin(), v.end(), array.buffer()); + + return array; +} } // namespace neon } // namespace test } // namespace arm_compute |