1 files changed, 14 insertions, 10 deletions
diff --git a/src/backends/backendsCommon/test/layerTests/MultiplicationTestImpl.cpp b/src/backends/backendsCommon/test/layerTests/MultiplicationTestImpl.cpp
index b8dc5f5667..c7b082183c 100644
--- a/src/backends/backendsCommon/test/layerTests/MultiplicationTestImpl.cpp
+++ b/src/backends/backendsCommon/test/layerTests/MultiplicationTestImpl.cpp
@@ -545,10 +545,11 @@ LayerTestResult<float,4> CompareMultiplicationTest(
     inputTensorInfo1 = armnn::TensorInfo(4, shape, armnn::DataType::Float32);
     outputTensorInfo = armnn::TensorInfo(4, shape, armnn::DataType::Float32);
 
-    LayerTestResult<float,4> comparisonResult(outputTensorInfo);
+    auto input0 = MakeRandomTensor<float>(inputTensorInfo0, 803506992);
+    auto input1 = MakeRandomTensor<float>(inputTensorInfo1, 54902257);
 
-    auto input0 = MakeRandomTensor<float, 4>(inputTensorInfo0, 803506992);
-    auto input1 = MakeRandomTensor<float, 4>(inputTensorInfo1, 54902257);
+    std::vector<float> actualOutput(outputTensorInfo.GetNumElements());
+    std::vector<float> expectedOutput(outputTensorInfo.GetNumElements());
 
     std::unique_ptr<armnn::ITensorHandle> inputHandle0 = tensorHandleFactory.CreateTensorHandle(inputTensorInfo0);
     std::unique_ptr<armnn::ITensorHandle> inputHandle1 = tensorHandleFactory.CreateTensorHandle(inputTensorInfo1);
@@ -580,17 +581,20 @@ LayerTestResult<float,4> CompareMultiplicationTest(
     inputHandle1Ref->Allocate();
     outputHandleRef->Allocate();
 
-    CopyDataToITensorHandle(inputHandle0.get(), input0.origin());
-    CopyDataToITensorHandle(inputHandle1.get(), input1.origin());
-    CopyDataToITensorHandle(inputHandle0Ref.get(), input0.origin());
-    CopyDataToITensorHandle(inputHandle1Ref.get(), input1.origin());
+    CopyDataToITensorHandle(inputHandle0.get(), input0.data());
+    CopyDataToITensorHandle(inputHandle1.get(), input1.data());
+    CopyDataToITensorHandle(inputHandle0Ref.get(), input0.data());
+    CopyDataToITensorHandle(inputHandle1Ref.get(), input1.data());
 
     workload->PostAllocationConfigure();
     workload->Execute();
     workloadRef->PostAllocationConfigure();
     workloadRef->Execute();
-    CopyDataFromITensorHandle(comparisonResult.output.origin(), outputHandle.get());
-    CopyDataFromITensorHandle(comparisonResult.outputExpected.origin(), outputHandleRef.get());
+    CopyDataFromITensorHandle(actualOutput.data(), outputHandle.get());
+    CopyDataFromITensorHandle(expectedOutput.data(), outputHandleRef.get());
 
-    return comparisonResult;
+    return LayerTestResult<float, 4>(actualOutput,
+                                     expectedOutput,
+                                     outputHandle->GetShape(),
+                                     outputTensorInfo.GetShape());
 }