diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2020-08-28 10:38:53 +0100 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2020-08-28 10:38:53 +0100 |
commit | 89de3b43a558710c7b1db2f4409b515d2fd94f0b (patch) | |
tree | b8c13b9ec9be56b3cef60e661885e09b5e9edadc /src/armnn/test/UnitTests.hpp | |
parent | 56785c75037ed0cd377851616634b3129713394b (diff) | |
download | armnn-89de3b43a558710c7b1db2f4409b515d2fd94f0b.tar.gz |
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 <sadik.armagan@arm.com>
Change-Id: I1a702941890034a45029c014c8b11e185f45a807
Diffstat (limited to 'src/armnn/test/UnitTests.hpp')
-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); } |