From 976ad3e257c4a68e809f5a7fe3b7a99bb3f1fba4 Mon Sep 17 00:00:00 2001 From: Ellen Norris-Thompson 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 Change-Id: Iaf4c2d5f4183ea54799bedae20950b23b914a727 --- test/Lstm.hpp | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'test/Lstm.hpp') diff --git a/test/Lstm.hpp b/test/Lstm.hpp index 3d9bf77f..6032f1c2 100644 --- a/test/Lstm.hpp +++ b/test/Lstm.hpp @@ -372,12 +372,12 @@ void LstmTestImpl(const hidl_vec& inputDimensions, AddPoolAndSetData(cellStateInValue.size(), request, cellStateInValue.data()); // add memory for the outputs - AddPoolAndGetData(scratchBufferValue.size(), request); - android::sp outputStateOutMemory = AddPoolAndGetData(outputStateOutValue.size(), request); + AddPoolAndGetData(scratchBufferValue.size(), request); + android::sp outputStateOutMemory = AddPoolAndGetData(outputStateOutValue.size(), request); float* outputStateOutData = static_cast(static_cast(outputStateOutMemory->getPointer())); - android::sp cellStateOutMemory = AddPoolAndGetData(cellStateOutValue.size(), request); + android::sp cellStateOutMemory = AddPoolAndGetData(cellStateOutValue.size(), request); float* cellStateOutData = static_cast(static_cast(cellStateOutMemory->getPointer())); - android::sp outputMemory = AddPoolAndGetData(outputValue.size(), request); + android::sp outputMemory = AddPoolAndGetData(outputValue.size(), request); float* outputData = static_cast(static_cast(outputMemory->getPointer())); // make the prepared model and run the execution -- cgit v1.2.1