aboutsummaryrefslogtreecommitdiff
path: root/src/backends/backendsCommon/test/layerTests/ArgMinMaxTestImpl.cpp
diff options
context:
space:
mode:
authorSadik Armagan <sadik.armagan@arm.com>2020-08-28 10:38:53 +0100
committerSadik Armagan <sadik.armagan@arm.com>2020-08-28 10:38:53 +0100
commit89de3b43a558710c7b1db2f4409b515d2fd94f0b (patch)
treeb8c13b9ec9be56b3cef60e661885e09b5e9edadc /src/backends/backendsCommon/test/layerTests/ArgMinMaxTestImpl.cpp
parent56785c75037ed0cd377851616634b3129713394b (diff)
downloadarmnn-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/backends/backendsCommon/test/layerTests/ArgMinMaxTestImpl.cpp')
-rw-r--r--src/backends/backendsCommon/test/layerTests/ArgMinMaxTestImpl.cpp86
1 files changed, 43 insertions, 43 deletions
diff --git a/src/backends/backendsCommon/test/layerTests/ArgMinMaxTestImpl.cpp b/src/backends/backendsCommon/test/layerTests/ArgMinMaxTestImpl.cpp
index 153d7269f7..d63cc04e99 100644
--- a/src/backends/backendsCommon/test/layerTests/ArgMinMaxTestImpl.cpp
+++ b/src/backends/backendsCommon/test/layerTests/ArgMinMaxTestImpl.cpp
@@ -19,7 +19,7 @@ template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
LayerTestResult<int32_t, 3> ArgMinMaxTestCommon(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr&,
- armnn::ITensorHandleFactory* tensorHandleFactory,
+ const armnn::ITensorHandleFactory& tensorHandleFactory,
armnn::ArgMinMaxFunction argMinMaxFunction,
const armnn::TensorInfo inputTensorInfo,
const armnn::TensorInfo outputTensorInfo,
@@ -32,8 +32,8 @@ LayerTestResult<int32_t, 3> ArgMinMaxTestCommon(
LayerTestResult<int32_t, 3> result(outputTensorInfo);
result.outputExpected = MakeTensor<int32_t, 3>(outputTensorInfo, outputData);
- std::unique_ptr<armnn::ITensorHandle> inputHandle = tensorHandleFactory->CreateTensorHandle(inputTensorInfo);
- std::unique_ptr<armnn::ITensorHandle> outputHandle = tensorHandleFactory->CreateTensorHandle(outputTensorInfo);
+ std::unique_ptr<armnn::ITensorHandle> inputHandle = tensorHandleFactory.CreateTensorHandle(inputTensorInfo);
+ std::unique_ptr<armnn::ITensorHandle> outputHandle = tensorHandleFactory.CreateTensorHandle(outputTensorInfo);
armnn::ArgMinMaxQueueDescriptor descriptor;
descriptor.m_Parameters.m_Function = argMinMaxFunction;
@@ -64,7 +64,7 @@ template<armnn::DataType ArmnnType, typename T>
LayerTestResult<int32_t, 3> ArgMaxSimpleTest(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory)
+ const armnn::ITensorHandleFactory& tensorHandleFactory)
{
const armnn::TensorShape inputShape{ 1, 1, 1, 5 };
const armnn::TensorShape outputShape{ 1, 1, 1 };
@@ -92,7 +92,7 @@ template<armnn::DataType ArmnnType, typename T>
LayerTestResult<int32_t, 3> ArgMinSimpleTest(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory)
+ const armnn::ITensorHandleFactory& tensorHandleFactory)
{
const armnn::TensorShape inputShape{ 1, 1, 1, 5 };
const armnn::TensorShape outputShape{ 1, 1, 1 };
@@ -120,7 +120,7 @@ template<armnn::DataType ArmnnType, typename T>
LayerTestResult<int32_t, 3> ArgMinChannelTest(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory)
+ const armnn::ITensorHandleFactory& tensorHandleFactory)
{
const armnn::TensorShape inputShape{ 1, 3, 2, 4};
const armnn::TensorShape outputShape{ 1, 2, 4 };
@@ -156,7 +156,7 @@ template<armnn::DataType ArmnnType, typename T>
LayerTestResult<int32_t, 3> ArgMaxChannelTest(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory)
+ const armnn::ITensorHandleFactory& tensorHandleFactory)
{
const armnn::TensorShape inputShape{ 1, 3, 2, 4};
const armnn::TensorShape outputShape{ 1, 2, 4 };
@@ -192,7 +192,7 @@ template<armnn::DataType ArmnnType, typename T>
LayerTestResult<int32_t, 3> ArgMaxHeightTest(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory)
+ const armnn::ITensorHandleFactory& tensorHandleFactory)
{
const armnn::TensorShape inputShape{ 1, 3, 2, 4};
const armnn::TensorShape outputShape{ 1, 3, 4 };
@@ -228,7 +228,7 @@ template<armnn::DataType ArmnnType, typename T>
LayerTestResult<int32_t, 3> ArgMinWidthTest(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory)
+ const armnn::ITensorHandleFactory& tensorHandleFactory)
{
const armnn::TensorShape inputShape{ 1, 3, 2, 4};
const armnn::TensorShape outputShape{ 1, 3, 2 };
@@ -267,202 +267,202 @@ template LayerTestResult<int32_t, 3>
ArgMaxSimpleTest<armnn::DataType::Float32>(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory);
+ const armnn::ITensorHandleFactory& tensorHandleFactory);
template LayerTestResult<int32_t, 3>
ArgMaxSimpleTest<armnn::DataType::Float16>(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory);
+ const armnn::ITensorHandleFactory& tensorHandleFactory);
template LayerTestResult<int32_t, 3>
ArgMaxSimpleTest<armnn::DataType::QAsymmS8>(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory);
+ const armnn::ITensorHandleFactory& tensorHandleFactory);
template LayerTestResult<int32_t, 3>
ArgMaxSimpleTest<armnn::DataType::QAsymmU8>(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory);
+ const armnn::ITensorHandleFactory& tensorHandleFactory);
template LayerTestResult<int32_t, 3>
ArgMaxSimpleTest<armnn::DataType::QSymmS16>(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory);
+ const armnn::ITensorHandleFactory& tensorHandleFactory);
template LayerTestResult<int32_t, 3>
ArgMaxSimpleTest<armnn::DataType::Signed32>(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory);
+ const armnn::ITensorHandleFactory& tensorHandleFactory);
template LayerTestResult<int32_t, 3>
ArgMinSimpleTest<armnn::DataType::Float32>(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory);
+ const armnn::ITensorHandleFactory& tensorHandleFactory);
template LayerTestResult<int32_t, 3>
ArgMinSimpleTest<armnn::DataType::Float16>(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory);
+ const armnn::ITensorHandleFactory& tensorHandleFactory);
template LayerTestResult<int32_t, 3>
ArgMinSimpleTest<armnn::DataType::QAsymmS8>(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory);
+ const armnn::ITensorHandleFactory& tensorHandleFactory);
template LayerTestResult<int32_t, 3>
ArgMinSimpleTest<armnn::DataType::QAsymmU8>(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory);
+ const armnn::ITensorHandleFactory& tensorHandleFactory);
template LayerTestResult<int32_t, 3>
ArgMinSimpleTest<armnn::DataType::QSymmS16>(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory);
+ const armnn::ITensorHandleFactory& tensorHandleFactory);
template LayerTestResult<int32_t, 3>
ArgMinSimpleTest<armnn::DataType::Signed32>(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory);
+ const armnn::ITensorHandleFactory& tensorHandleFactory);
template LayerTestResult<int32_t, 3>
ArgMinChannelTest<armnn::DataType::Float32>(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory);
+ const armnn::ITensorHandleFactory& tensorHandleFactory);
template LayerTestResult<int32_t, 3>
ArgMinChannelTest<armnn::DataType::Float16>(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory);
+ const armnn::ITensorHandleFactory& tensorHandleFactory);
template LayerTestResult<int32_t, 3>
ArgMinChannelTest<armnn::DataType::QAsymmS8>(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory);
+ const armnn::ITensorHandleFactory& tensorHandleFactory);
template LayerTestResult<int32_t, 3>
ArgMinChannelTest<armnn::DataType::QAsymmU8>(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory);
+ const armnn::ITensorHandleFactory& tensorHandleFactory);
template LayerTestResult<int32_t, 3>
ArgMinChannelTest<armnn::DataType::QSymmS16>(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory);
+ const armnn::ITensorHandleFactory& tensorHandleFactory);
template LayerTestResult<int32_t, 3>
ArgMinChannelTest<armnn::DataType::Signed32>(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory);
+ const armnn::ITensorHandleFactory& tensorHandleFactory);
template LayerTestResult<int32_t, 3>
ArgMaxChannelTest<armnn::DataType::Float32>(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory);
+ const armnn::ITensorHandleFactory& tensorHandleFactory);
template LayerTestResult<int32_t, 3>
ArgMaxChannelTest<armnn::DataType::Float16>(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory);
+ const armnn::ITensorHandleFactory& tensorHandleFactory);
template LayerTestResult<int32_t, 3>
ArgMaxChannelTest<armnn::DataType::QAsymmS8>(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory);
+ const armnn::ITensorHandleFactory& tensorHandleFactory);
template LayerTestResult<int32_t, 3>
ArgMaxChannelTest<armnn::DataType::QAsymmU8>(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory);
+ const armnn::ITensorHandleFactory& tensorHandleFactory);
template LayerTestResult<int32_t, 3>
ArgMaxChannelTest<armnn::DataType::QSymmS16>(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory);
+ const armnn::ITensorHandleFactory& tensorHandleFactory);
template LayerTestResult<int32_t, 3>
ArgMaxChannelTest<armnn::DataType::Signed32>(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory);
+ const armnn::ITensorHandleFactory& tensorHandleFactory);
template LayerTestResult<int32_t, 3>
ArgMaxHeightTest<armnn::DataType::Float32>(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory);
+ const armnn::ITensorHandleFactory& tensorHandleFactory);
template LayerTestResult<int32_t, 3>
ArgMaxHeightTest<armnn::DataType::Float16>(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory);
+ const armnn::ITensorHandleFactory& tensorHandleFactory);
template LayerTestResult<int32_t, 3>
ArgMaxHeightTest<armnn::DataType::Signed32>(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory);
+ const armnn::ITensorHandleFactory& tensorHandleFactory);
template LayerTestResult<int32_t, 3>
ArgMaxHeightTest<armnn::DataType::QAsymmS8>(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory);
+ const armnn::ITensorHandleFactory& tensorHandleFactory);
template LayerTestResult<int32_t, 3>
ArgMaxHeightTest<armnn::DataType::QAsymmU8>(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory);
+ const armnn::ITensorHandleFactory& tensorHandleFactory);
template LayerTestResult<int32_t, 3>
ArgMinWidthTest<armnn::DataType::Float32>(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory);
+ const armnn::ITensorHandleFactory& tensorHandleFactory);
template LayerTestResult<int32_t, 3>
ArgMinWidthTest<armnn::DataType::Float16>(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory);
+ const armnn::ITensorHandleFactory& tensorHandleFactory);
template LayerTestResult<int32_t, 3>
ArgMinWidthTest<armnn::DataType::Signed32>(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory);
+ const armnn::ITensorHandleFactory& tensorHandleFactory);
template LayerTestResult<int32_t, 3>
ArgMinWidthTest<armnn::DataType::QAsymmS8>(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory);
+ const armnn::ITensorHandleFactory& tensorHandleFactory);
template LayerTestResult<int32_t, 3>
ArgMinWidthTest<armnn::DataType::QAsymmU8>(
armnn::IWorkloadFactory& workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
- armnn::ITensorHandleFactory* tensorHandleFactory);
+ const armnn::ITensorHandleFactory& tensorHandleFactory);