diff options
author | Finn Williams <Finn.Williams@arm.com> | 2020-08-27 16:15:20 +0100 |
---|---|---|
committer | finn.williams <finn.williams@arm.com> | 2020-08-31 15:54:38 +0000 |
commit | 826a543547f325bfac7da507d19a55af9c138a50 (patch) | |
tree | 84b380534fac9553e35da37f84ed93bc3bf10fd3 /src/backends/backendsCommon/test/layerTests/MeanTestImpl.hpp | |
parent | c43de6a6a4af212d599824fc783871c440a9fda8 (diff) | |
download | armnn-826a543547f325bfac7da507d19a55af9c138a50.tar.gz |
IVGCVSW-5253 Use CreateTensorHandle() function from TensorHandleFactory in the tests for layers M-P
Signed-off-by: Finn Williams <Finn.Williams@arm.com>
Change-Id: I324eee7d750e30f714e0d346b7da7b69866ff935
Diffstat (limited to 'src/backends/backendsCommon/test/layerTests/MeanTestImpl.hpp')
-rw-r--r-- | src/backends/backendsCommon/test/layerTests/MeanTestImpl.hpp | 45 |
1 files changed, 27 insertions, 18 deletions
diff --git a/src/backends/backendsCommon/test/layerTests/MeanTestImpl.hpp b/src/backends/backendsCommon/test/layerTests/MeanTestImpl.hpp index 283bfd9502..ba827b1860 100644 --- a/src/backends/backendsCommon/test/layerTests/MeanTestImpl.hpp +++ b/src/backends/backendsCommon/test/layerTests/MeanTestImpl.hpp @@ -19,6 +19,7 @@ template<armnn::DataType ArmnnType, typename T, std::size_t InputDim, std::size_ LayerTestResult<T, OutputDim> MeanTestHelper( armnn::IWorkloadFactory& workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, + const armnn::ITensorHandleFactory& tensorHandleFactory, const unsigned int* inputShape, const std::vector<float>& inputData, const std::vector<unsigned int>& axis, @@ -45,10 +46,8 @@ LayerTestResult<T, OutputDim> MeanTestHelper( result.outputExpected = MakeTensor<T, OutputDim>( outputTensorInfo, ConvertToDataType<ArmnnType>(outputData, outputTensorInfo)); - ARMNN_NO_DEPRECATE_WARN_BEGIN - std::unique_ptr<armnn::ITensorHandle> inputHandle = workloadFactory.CreateTensorHandle(inputTensorInfo); - std::unique_ptr<armnn::ITensorHandle> outputHandle = workloadFactory.CreateTensorHandle(outputTensorInfo); - ARMNN_NO_DEPRECATE_WARN_END + std::unique_ptr<armnn::ITensorHandle> inputHandle = tensorHandleFactory.CreateTensorHandle(inputTensorInfo); + std::unique_ptr<armnn::ITensorHandle> outputHandle = tensorHandleFactory.CreateTensorHandle(outputTensorInfo); armnn::MeanQueueDescriptor data; data.m_Parameters.m_Axis = axis; @@ -77,7 +76,8 @@ LayerTestResult<T, OutputDim> MeanTestHelper( template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> LayerTestResult<T, 1> MeanSimpleTest( armnn::IWorkloadFactory& workloadFactory, - const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager) + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, + const armnn::ITensorHandleFactory& tensorHandleFactory) { const unsigned int inputShape[] = { 3, 2 }; const unsigned int outputShape[] = { 1 }; @@ -86,13 +86,14 @@ LayerTestResult<T, 1> MeanSimpleTest( std::vector<float> output({ 2.5f }); return MeanTestHelper<ArmnnType, T, 2, 1>( - workloadFactory, memoryManager, inputShape, input, {}, false, outputShape, output); + workloadFactory, memoryManager, tensorHandleFactory, inputShape, input, {}, false, outputShape, output); } template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> LayerTestResult<T, 3> MeanSimpleAxisTest( armnn::IWorkloadFactory& workloadFactory, - const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager) + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, + const armnn::ITensorHandleFactory& tensorHandleFactory) { const unsigned int inputShape[] = { 2, 3, 1, 2 }; const unsigned int outputShape[] = { 3, 1, 2 }; @@ -101,13 +102,14 @@ LayerTestResult<T, 3> MeanSimpleAxisTest( std::vector<float> output({ 1.5f, 2.5f, 3.5f, 4.5f, 5.5f, 6.5f }); return MeanTestHelper<ArmnnType, T, 4, 3>( - workloadFactory, memoryManager, inputShape, input, { 0 }, false, outputShape, output); + workloadFactory, memoryManager, tensorHandleFactory, inputShape, input, { 0 }, false, outputShape, output); } template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> LayerTestResult<T, 4> MeanKeepDimsTest( armnn::IWorkloadFactory& workloadFactory, - const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager) + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, + const armnn::ITensorHandleFactory& tensorHandleFactory) { const unsigned int inputShape[] = { 1, 1, 3, 2 }; const unsigned int outputShape[] = { 1, 1, 1, 2 }; @@ -116,13 +118,14 @@ LayerTestResult<T, 4> MeanKeepDimsTest( std::vector<float> output({ 2.5f, 2.5f }); return MeanTestHelper<ArmnnType, T, 4, 4>( - workloadFactory, memoryManager, inputShape, input, { 2 }, true, outputShape, output); + workloadFactory, memoryManager, tensorHandleFactory, inputShape, input, { 2 }, true, outputShape, output); } template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> LayerTestResult<T, 4> MeanMultipleDimsTest( armnn::IWorkloadFactory& workloadFactory, - const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager) + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, + const armnn::ITensorHandleFactory& tensorHandleFactory) { const unsigned int inputShape[] = { 2, 3, 1, 2 }; const unsigned int outputShape[] = { 1, 3, 1, 1 }; @@ -131,13 +134,15 @@ LayerTestResult<T, 4> MeanMultipleDimsTest( std::vector<float> output({ 2.0f, 4.0f, 6.0f }); return MeanTestHelper<ArmnnType, T, 4, 4>( - workloadFactory, memoryManager, inputShape, input, { 0, 3 }, true, outputShape, output); + workloadFactory, memoryManager, tensorHandleFactory, + inputShape, input, { 0, 3 }, true, outputShape, output); } template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> LayerTestResult<T, 1> MeanVts1Test( armnn::IWorkloadFactory& workloadFactory, - const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager) + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, + const armnn::ITensorHandleFactory& tensorHandleFactory) { const unsigned int inputShape[] = { 4, 3, 2 }; const unsigned int outputShape[] = { 2 }; @@ -147,13 +152,15 @@ LayerTestResult<T, 1> MeanVts1Test( std::vector<float> output({ 12.0f, 13.0f }); return MeanTestHelper<ArmnnType, T, 3, 1>( - workloadFactory, memoryManager, inputShape, input, { 0, 1 }, false, outputShape, output); + workloadFactory, memoryManager, tensorHandleFactory, + inputShape, input, { 0, 1 }, false, outputShape, output); } template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> LayerTestResult<T, 3> MeanVts2Test( armnn::IWorkloadFactory& workloadFactory, - const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager) + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, + const armnn::ITensorHandleFactory& tensorHandleFactory) { const unsigned int inputShape[] = { 4, 3, 2 }; const unsigned int outputShape[] = { 1, 3, 1 }; @@ -163,13 +170,15 @@ LayerTestResult<T, 3> MeanVts2Test( std::vector<float> output({ 10.5f, 12.5f, 14.5f }); return MeanTestHelper<ArmnnType, T, 3, 3>( - workloadFactory, memoryManager, inputShape, input, { 0, 2 }, true, outputShape, output); + workloadFactory, memoryManager, tensorHandleFactory, + inputShape, input, { 0, 2 }, true, outputShape, output); } template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> LayerTestResult<T, 3> MeanVts3Test( armnn::IWorkloadFactory& workloadFactory, - const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager) + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, + const armnn::ITensorHandleFactory& tensorHandleFactory) { const unsigned int inputShape[] = { 1, 2, 2, 1 }; const unsigned int outputShape[] = { 1, 2, 1 }; @@ -178,5 +187,5 @@ LayerTestResult<T, 3> MeanVts3Test( std::vector<float> output({ 1.5f, 3.5f }); return MeanTestHelper<ArmnnType, T, 4, 3>( - workloadFactory, memoryManager, inputShape, input, { 2 }, false, outputShape, output); + workloadFactory, memoryManager, tensorHandleFactory, inputShape, input, { 2 }, false, outputShape, output); } |