diff options
author | Finn Williams <Finn.Williams@arm.com> | 2020-08-27 10:37:36 +0100 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2020-08-31 09:26:33 +0000 |
commit | 7faf9a88637afae76373a439d7eca3507b77c98c (patch) | |
tree | 9ef1b69e89c6a85de89f6d464f23754d7591ec86 /src/backends/backendsCommon/test/layerTests/FullyConnectedTestImpl.hpp | |
parent | 5e71984463ee83e75ddd5aee1f890d49b2bd1756 (diff) | |
download | armnn-7faf9a88637afae76373a439d7eca3507b77c98c.tar.gz |
IVGCVSW-5250 Remove CreateTensorHandle in the test for layers between E-F
* Refactored Floor and FullyConnected tests
Signed-off-by: Finn Williams <Finn.Williams@arm.com>
Change-Id: Iad87254e638bdcb5d7b334b16ec87a0c981e48a0
Diffstat (limited to 'src/backends/backendsCommon/test/layerTests/FullyConnectedTestImpl.hpp')
-rw-r--r-- | src/backends/backendsCommon/test/layerTests/FullyConnectedTestImpl.hpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/backends/backendsCommon/test/layerTests/FullyConnectedTestImpl.hpp b/src/backends/backendsCommon/test/layerTests/FullyConnectedTestImpl.hpp index d181557d3f..c2d53a5178 100644 --- a/src/backends/backendsCommon/test/layerTests/FullyConnectedTestImpl.hpp +++ b/src/backends/backendsCommon/test/layerTests/FullyConnectedTestImpl.hpp @@ -16,15 +16,18 @@ template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>> LayerTestResult<T, 2> FullyConnectedTest( armnn::IWorkloadFactory& workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, + const armnn::ITensorHandleFactory& tensorHandleFactory, bool biasEnabled); LayerTestResult<float, 2> FullyConnectedFloat32Test( armnn::IWorkloadFactory& workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, + const armnn::ITensorHandleFactory& tensorHandleFactory, bool biasEnabled, bool transposeWeights); LayerTestResult<float, 2> FullyConnectedLargeTest( armnn::IWorkloadFactory& workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, + const armnn::ITensorHandleFactory& tensorHandleFactory, bool transposeWeights); |