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/FullyConnected.cpp | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'test/FullyConnected.cpp') diff --git a/test/FullyConnected.cpp b/test/FullyConnected.cpp index de885153..ec4fcbda 100644 --- a/test/FullyConnected.cpp +++ b/test/FullyConnected.cpp @@ -70,10 +70,10 @@ BOOST_AUTO_TEST_CASE(FullyConnected) // set the input data (matching source test) float indata[] = {2, 32, 16}; - AddPoolAndSetData(3, request, indata); + AddPoolAndSetData(3, request, indata); // add memory for the output - android::sp outMemory = AddPoolAndGetData(1, request); + android::sp outMemory = AddPoolAndGetData(1, request); float* outdata = static_cast(static_cast(outMemory->getPointer())); // run the execution @@ -153,7 +153,7 @@ BOOST_AUTO_TEST_CASE(TestFullyConnected4dInput) AddPoolAndSetData(8, request, indata); // add memory for the output - android::sp outMemory = AddPoolAndGetData(8, request); + android::sp outMemory = AddPoolAndGetData(8, request); float* outdata = static_cast(static_cast(outMemory->getPointer())); // run the execution @@ -240,7 +240,7 @@ BOOST_AUTO_TEST_CASE(TestFullyConnected4dInputReshape) AddPoolAndSetData(8, request, indata); // add memory for the output - android::sp outMemory = AddPoolAndGetData(8, request); + android::sp outMemory = AddPoolAndGetData(8, request); float* outdata = static_cast(static_cast(outMemory->getPointer())); // run the execution -- cgit v1.2.1