diff options
author | Ellen Norris-Thompson <ellen.norris-thompson@arm.com> | 2019-08-21 15:21:14 +0100 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-08-22 08:45:13 +0100 |
commit | 976ad3e257c4a68e809f5a7fe3b7a99bb3f1fba4 (patch) | |
tree | 683a03f831a8d4706ab2a34d155b4b9c3bd0fc44 /test/Lstm.hpp | |
parent | 1ba9989c4d4fe152d2cb02c83c2da1c18b6aaf37 (diff) | |
download | android-nn-driver-976ad3e257c4a68e809f5a7fe3b7a99bb3f1fba4.tar.gz |
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
Diffstat (limited to 'test/Lstm.hpp')
-rw-r--r-- | test/Lstm.hpp | 8 |
1 files changed, 4 insertions, 4 deletions
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<uint32_t>& inputDimensions, AddPoolAndSetData(cellStateInValue.size(), request, cellStateInValue.data()); // add memory for the outputs - AddPoolAndGetData(scratchBufferValue.size(), request); - android::sp<IMemory> outputStateOutMemory = AddPoolAndGetData(outputStateOutValue.size(), request); + AddPoolAndGetData<float>(scratchBufferValue.size(), request); + android::sp<IMemory> outputStateOutMemory = AddPoolAndGetData<float>(outputStateOutValue.size(), request); float* outputStateOutData = static_cast<float*>(static_cast<void*>(outputStateOutMemory->getPointer())); - android::sp<IMemory> cellStateOutMemory = AddPoolAndGetData(cellStateOutValue.size(), request); + android::sp<IMemory> cellStateOutMemory = AddPoolAndGetData<float>(cellStateOutValue.size(), request); float* cellStateOutData = static_cast<float*>(static_cast<void*>(cellStateOutMemory->getPointer())); - android::sp<IMemory> outputMemory = AddPoolAndGetData(outputValue.size(), request); + android::sp<IMemory> outputMemory = AddPoolAndGetData<float>(outputValue.size(), request); float* outputData = static_cast<float*>(static_cast<void*>(outputMemory->getPointer())); // make the prepared model and run the execution |