diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2021-02-12 17:16:42 +0000 |
---|---|---|
committer | Kevin May <kevin.may@arm.com> | 2021-02-15 16:50:24 +0000 |
commit | 188675fd7e7239f56212d72f8ac10598de87e76a (patch) | |
tree | 611876f7c43f81c88b8dec19ae86afcb6d6a1a64 /test/Lstm.hpp | |
parent | cd639c98c347c1addae8c65eb1f4d8e75b468fdb (diff) | |
download | android-nn-driver-188675fd7e7239f56212d72f8ac10598de87e76a.tar.gz |
IVGCVSW-5665 Basic NN Driver support for next OS Version
Signed-off-by: Kevin May <kevin.may@arm.com>
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I1e1db52322092c6b1b7ac6183c3adc90aabcec24
Diffstat (limited to 'test/Lstm.hpp')
-rw-r--r-- | test/Lstm.hpp | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/test/Lstm.hpp b/test/Lstm.hpp index 21056c3d..2cb3c264 100644 --- a/test/Lstm.hpp +++ b/test/Lstm.hpp @@ -15,6 +15,11 @@ using ArmnnDriver = armnn_driver::ArmnnDriver; using DriverOptions = armnn_driver::DriverOptions; +using RequestArgument = V1_0::RequestArgument; + +#ifdef ARMNN_ANDROID_S +#include <nnapi/Types.h> +#endif using namespace driverTestHelpers; using namespace android::hardware; @@ -25,7 +30,7 @@ namespace template<typename T> RequestArgument CreateRequestArgument(const std::vector<T>& value, unsigned int poolIndex) { - DataLocation inputInloc = {}; + V1_0::DataLocation inputInloc = {}; inputInloc.poolIndex = poolIndex; inputInloc.offset = 0; inputInloc.length = value.size() * sizeof(T); |