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 <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<T, 2> FullyConnectedTest( 17 armnn::IWorkloadFactory& workloadFactory, 18 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 19 const armnn::ITensorHandleFactory& tensorHandleFactory, 20 bool biasEnabled, 21 bool constantWeights); 22 23 LayerTestResult<float, 2> FullyConnectedFloat32Test( 24 armnn::IWorkloadFactory& workloadFactory, 25 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 26 const armnn::ITensorHandleFactory& tensorHandleFactory, 27 bool biasEnabled, 28 bool transposeWeights); 29 30 LayerTestResult<float, 2> FullyConnectedLargeTest( 31 armnn::IWorkloadFactory& workloadFactory, 32 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 33 const armnn::ITensorHandleFactory& tensorHandleFactory, 34 bool transposeWeights); 35