From 89de3b43a558710c7b1db2f4409b515d2fd94f0b Mon Sep 17 00:00:00 2001 From: Sadik Armagan Date: Fri, 28 Aug 2020 10:38:53 +0100 Subject: IVGCVSW-4979 'Remove CreateTensorHandle using WorkloadFactory in workload tests' * Small refactor in unit tests using TensorHandleFactory to use reference instead of pointer Signed-off-by: Sadik Armagan Change-Id: I1a702941890034a45029c014c8b11e185f45a807 --- src/armnn/test/UnitTests.hpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src/armnn/test') diff --git a/src/armnn/test/UnitTests.hpp b/src/armnn/test/UnitTests.hpp index c049f578fc..a77e25539a 100644 --- a/src/armnn/test/UnitTests.hpp +++ b/src/armnn/test/UnitTests.hpp @@ -88,7 +88,7 @@ void RunTestFunctionUsingTensorHandleFactory(const char* testName, TFuncPtr test auto tensorHandleFactory = WorkloadFactoryHelper::GetTensorHandleFactory(memoryManager); - auto testResult = (*testFunction)(workloadFactory, memoryManager, &tensorHandleFactory, args...); + auto testResult = (*testFunction)(workloadFactory, memoryManager, tensorHandleFactory, args...); CompareTestResultIfSupported(testName, testResult); armnn::ProfilerManager::GetInstance().RegisterProfiler(nullptr); @@ -136,7 +136,7 @@ void CompareRefTestFunctionUsingTensorHandleFactory(const char* testName, TFuncP RefWorkloadFactoryHelper::GetTensorHandleFactory(memoryManager); auto testResult = (*testFunction)( - workloadFactory, memoryManager, refWorkloadFactory, &tensorHandleFactory, &refTensorHandleFactory, args...); + workloadFactory, memoryManager, refWorkloadFactory, tensorHandleFactory, refTensorHandleFactory, args...); CompareTestResultIfSupported(testName, testResult); } -- cgit v1.2.1