From 976ad3e257c4a68e809f5a7fe3b7a99bb3f1fba4 Mon Sep 17 00:00:00 2001
From: Ellen Norris-Thompson <ellen.norris-thompson@arm.com>
Date: Wed, 21 Aug 2019 15:21:14 +0100
Subject: IVGCVSW-3613 Refactoring Android NN driver for QuantisedLstm test

 * Templated AddPoolAndGetData and AddPoolAndSetData to accept different types
 * Moved scale and offset to arguments of Add*Operand functions

Signed-off-by: Ellen Norris-Thompson <ellen.norris-thompson@arm.com>
Change-Id: Iaf4c2d5f4183ea54799bedae20950b23b914a727
---
 test/Convolution2D.hpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'test/Convolution2D.hpp')

diff --git a/test/Convolution2D.hpp b/test/Convolution2D.hpp
index 46b46549..180f57e2 100644
--- a/test/Convolution2D.hpp
+++ b/test/Convolution2D.hpp
@@ -102,7 +102,7 @@ void PaddingTestImpl(android::nn::PaddingScheme paddingScheme, bool fp16Enabled
     AddPoolAndSetData(6, request, indata);
 
     // add memory for the output
-    android::sp<IMemory> outMemory = AddPoolAndGetData(outSize, request);
+    android::sp<IMemory> outMemory = AddPoolAndGetData<float>(outSize, request);
     float* outdata = reinterpret_cast<float*>(static_cast<void*>(outMemory->getPointer()));
 
     // run the execution
-- 
cgit v1.2.1