1 // 2 // Copyright © 2017 Arm Ltd. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 6 #pragma once 7 8 #include <armnnTestUtils/LayerTestResult.hpp> 9 10 #include <armnn/Types.hpp> 11 12 #include <armnn/backends/IBackendInternal.hpp> 13 #include <armnn/backends/WorkloadFactory.hpp> 14 15 LayerTestResult<float, 4> SimpleNormalizationAcrossTest( 16 armnn::IWorkloadFactory& workloadFactory, 17 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 18 const armnn::ITensorHandleFactory& tensorHandleFactory); 19 20 LayerTestResult<float, 4> SimpleNormalizationWithinTest( 21 armnn::IWorkloadFactory& workloadFactory, 22 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 23 const armnn::ITensorHandleFactory& tensorHandleFactory); 24 25 LayerTestResult<float,4> SimpleNormalizationAcrossNhwcTest( 26 armnn::IWorkloadFactory& workloadFactory, 27 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 28 const armnn::ITensorHandleFactory& tensorHandleFactory); 29 30 LayerTestResult<float, 4> CompareNormalizationTest( 31 armnn::IWorkloadFactory& workloadFactory, 32 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 33 armnn::IWorkloadFactory& refWorkloadFactory, 34 const armnn::ITensorHandleFactory& tensorHandleFactory, 35 const armnn::ITensorHandleFactory& refTensorHandleFactory, 36 armnn::NormalizationAlgorithmChannel normChannel, 37 armnn::NormalizationAlgorithmMethod normMethod); 38 39 LayerTestResult<float, 4> AcrossChannelNormalizationTest( 40 armnn::IWorkloadFactory& workloadFactory, 41 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 42 const armnn::ITensorHandleFactory& tensorHandleFactory);