From 3537c2ca7ebf31c1673b9ec2bb0c17b0406bbae0 Mon Sep 17 00:00:00 2001
From: surmeh01 <surabhi.mehta@arm.com>
Date: Fri, 18 May 2018 16:31:43 +0100
Subject: Release 18.05

---
 src/armnn/backends/test/FullyConnectedTestImpl.hpp | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

(limited to 'src/armnn/backends/test/FullyConnectedTestImpl.hpp')

diff --git a/src/armnn/backends/test/FullyConnectedTestImpl.hpp b/src/armnn/backends/test/FullyConnectedTestImpl.hpp
index 479da3fabc..d2379ec10e 100644
--- a/src/armnn/backends/test/FullyConnectedTestImpl.hpp
+++ b/src/armnn/backends/test/FullyConnectedTestImpl.hpp
@@ -10,9 +10,9 @@ LayerTestResult<T, 2> SimpleFullyConnectedTestImpl(
     armnn::TensorInfo outputTensorInfo,
     armnn::TensorInfo weightsDesc,
     armnn::TensorInfo biasesDesc,
-    boost::multi_array<T, 2> weights,
-    boost::multi_array<B, 1> bias,
-    boost::multi_array<T, 4> input,
+    boost::multi_array<T, 2>& weights,
+    boost::multi_array<B, 1>& bias,
+    boost::multi_array<T, 4>& input,
     bool biasEnabled,
     bool transposeWeights)
 {
@@ -41,6 +41,7 @@ LayerTestResult<T, 2> SimpleFullyConnectedTestImpl(
     outputHandle->Allocate();
     CopyDataToITensorHandle(inputHandle.get(), &input[0][0][0][0]);
 
+    workloadFactory.Finalize();
     workload->Execute();
 
     CopyDataFromITensorHandle(&result.output[0][0], outputHandle.get());
-- 
cgit v1.2.1