aboutsummaryrefslogtreecommitdiff
path: root/src/backends/backendsCommon/test/layerTests/NormalizationTestImpl.hpp
diff options
context:
space:
mode:
authorFinn Williams <Finn.Williams@arm.com>2020-08-27 16:15:20 +0100
committerfinn.williams <finn.williams@arm.com>2020-08-31 15:54:38 +0000
commit826a543547f325bfac7da507d19a55af9c138a50 (patch)
tree84b380534fac9553e35da37f84ed93bc3bf10fd3 /src/backends/backendsCommon/test/layerTests/NormalizationTestImpl.hpp
parentc43de6a6a4af212d599824fc783871c440a9fda8 (diff)
downloadarmnn-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.hpp11
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);