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

1 files changed, 15 insertions, 14 deletions

diff --git a/test/1.1/Mean.cpp b/test/1.1/Mean.cpp
index 8c52d23d..c9a5a6d3 100644
--- a/test/1.1/Mean.cpp
+++ b/test/1.1/Mean.cpp
@@ -19,6 +19,7 @@ using namespace driverTestHelpers;
 using namespace armnn_driver;
 
 using HalPolicy = hal_1_1::HalPolicy;
+using RequestArgument = V1_0::RequestArgument;
 
 namespace
 {
@@ -61,22 +62,22 @@ void MeanTestImpl(const TestTensor& input,
     android::sp<V1_0::IPreparedModel> preparedModel = PrepareModel(model, *driver);
 
     // The request's memory pools will follow the same order as the inputs
-    DataLocation inLoc    = {};
-    inLoc.poolIndex       = 0;
-    inLoc.offset          = 0;
-    inLoc.length          = input.GetNumElements() * sizeof(float);
-    RequestArgument inArg = {};
-    inArg.location        = inLoc;
-    inArg.dimensions      = input.GetDimensions();
+    V1_0::DataLocation inLoc = {};
+    inLoc.poolIndex          = 0;
+    inLoc.offset             = 0;
+    inLoc.length             = input.GetNumElements() * sizeof(float);
+    RequestArgument inArg    = {};
+    inArg.location           = inLoc;
+    inArg.dimensions         = input.GetDimensions();
 
     // An additional memory pool is needed for the output
-    DataLocation outLoc    = {};
-    outLoc.poolIndex       = 1;
-    outLoc.offset          = 0;
-    outLoc.length          = expectedOutput.GetNumElements() * sizeof(float);
-    RequestArgument outArg = {};
-    outArg.location        = outLoc;
-    outArg.dimensions      = expectedOutput.GetDimensions();
+    V1_0::DataLocation outLoc = {};
+    outLoc.poolIndex          = 1;
+    outLoc.offset             = 0;
+    outLoc.length             = expectedOutput.GetNumElements() * sizeof(float);
+    RequestArgument outArg    = {};
+    outArg.location           = outLoc;
+    outArg.dimensions         = expectedOutput.GetDimensions();
 
     // Make the request based on the arguments
     V1_0::Request request = {};