blob: 4da1ad2606b610b679b6506cdcc7ced4962e9487 [file] [log] [blame]
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +01001//
2// Copyright © 2017 Arm Ltd. All rights reserved.
3// SPDX-License-Identifier: MIT
4//
5
6#pragma once
7
8#include "LayerTestResult.hpp"
9
10#include <armnn/Types.hpp>
11
Matteo Martincighe5b8eb92019-11-28 15:45:42 +000012#include <armnn/backends/IBackendInternal.hpp>
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +010013#include <backendsCommon/WorkloadFactory.hpp>
14
15LayerTestResult<float, 4> SimpleNormalizationAcrossTest(
16 armnn::IWorkloadFactory& workloadFactory,
17 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
18
19LayerTestResult<float, 4> SimpleNormalizationWithinTest(
20 armnn::IWorkloadFactory& workloadFactory,
21 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
22
23LayerTestResult<float,4> SimpleNormalizationAcrossNhwcTest(
24 armnn::IWorkloadFactory& workloadFactory,
25 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
26
27LayerTestResult<float, 4> CompareNormalizationTest(
28 armnn::IWorkloadFactory& workloadFactory,
29 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
30 armnn::IWorkloadFactory& refWorkloadFactory,
31 armnn::NormalizationAlgorithmChannel normChannel,
32 armnn::NormalizationAlgorithmMethod normMethod);