diff options
Diffstat (limited to 'src/armnn/test')
-rw-r--r-- | src/armnn/test/UnitTests.hpp | 4 |
1 files changed, 2 insertions, 2 deletions
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<FactoryType>::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); } |