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/NormalizationTestImpl.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/NormalizationTestImpl.hpp')
-rw-r--r-- | src/backends/backendsCommon/test/layerTests/NormalizationTestImpl.hpp | 11 |
1 files changed, 8 insertions, 3 deletions
diff --git a/src/backends/backendsCommon/test/layerTests/NormalizationTestImpl.hpp b/src/backends/backendsCommon/test/layerTests/NormalizationTestImpl.hpp index 4da1ad2606..3a276e8c4c 100644 --- a/src/backends/backendsCommon/test/layerTests/NormalizationTestImpl.hpp +++ b/src/backends/backendsCommon/test/layerTests/NormalizationTestImpl.hpp @@ -14,19 +14,24 @@ LayerTestResult<float, 4> SimpleNormalizationAcrossTest( armnn::IWorkloadFactory& workloadFactory, - const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager); + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, + const armnn::ITensorHandleFactory& tensorHandleFactory); LayerTestResult<float, 4> SimpleNormalizationWithinTest( armnn::IWorkloadFactory& workloadFactory, - const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager); + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, + const armnn::ITensorHandleFactory& tensorHandleFactory); LayerTestResult<float,4> SimpleNormalizationAcrossNhwcTest( armnn::IWorkloadFactory& workloadFactory, - const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager); + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, + const armnn::ITensorHandleFactory& tensorHandleFactory); LayerTestResult<float, 4> CompareNormalizationTest( armnn::IWorkloadFactory& workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, armnn::IWorkloadFactory& refWorkloadFactory, + const armnn::ITensorHandleFactory& tensorHandleFactory, + const armnn::ITensorHandleFactory& refTensorHandleFactory, armnn::NormalizationAlgorithmChannel normChannel, armnn::NormalizationAlgorithmMethod normMethod); |