1 // 2 // Copyright © 2021 Arm Ltd and Contributors. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 6 #pragma once 7 8 #include <armnnTestUtils/LayerTestResult.hpp> 9 10 #include <ResolveType.hpp> 11 12 #include <armnn/backends/IBackendInternal.hpp> 13 #include <armnn/backends/WorkloadFactory.hpp> 14 15 template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> 16 LayerTestResult<float, 4> ReduceProdSimpleTest( 17 armnn::IWorkloadFactory& workloadFactory, 18 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 19 const armnn::ITensorHandleFactory& tensorHandleFactory); 20 21 template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> 22 LayerTestResult<float, 4> ReduceProdSingleAxisTest1( 23 armnn::IWorkloadFactory& workloadFactory, 24 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 25 const armnn::ITensorHandleFactory& tensorHandleFactory); 26 27 template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> 28 LayerTestResult<float, 4> ReduceProdSingleAxisTest2( 29 armnn::IWorkloadFactory& workloadFactory, 30 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 31 const armnn::ITensorHandleFactory& tensorHandleFactory); 32 33 template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> 34 LayerTestResult<float, 4> ReduceProdSingleAxisTest3( 35 armnn::IWorkloadFactory& workloadFactory, 36 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 37 const armnn::ITensorHandleFactory& tensorHandleFactory); 38 39 template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> 40 LayerTestResult<float, 4> ReduceProdMultipleAxisTest( 41 armnn::IWorkloadFactory& workloadFactory, 42 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 43 const armnn::ITensorHandleFactory& tensorHandleFactory); 44