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/FloorTestImpl.cpp | |
| 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/FloorTestImpl.cpp')
| -rw-r--r-- | src/backends/backendsCommon/test/layerTests/FloorTestImpl.cpp | 18 |
1 files changed, 10 insertions, 8 deletions
diff --git a/src/backends/backendsCommon/test/layerTests/FloorTestImpl.cpp b/src/backends/backendsCommon/test/layerTests/FloorTestImpl.cpp index 7072698374..46f384266b 100644 --- a/src/backends/backendsCommon/test/layerTests/FloorTestImpl.cpp +++ b/src/backends/backendsCommon/test/layerTests/FloorTestImpl.cpp @@ -14,7 +14,8 @@ template<armnn::DataType ArmnnType, typename T> LayerTestResult<T, 4> SimpleFloorTest( armnn::IWorkloadFactory& workloadFactory, - const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager) + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, + const armnn::ITensorHandleFactory& tensorHandleFactory) { IgnoreUnused(memoryManager); armnn::TensorInfo inputTensorInfo({1, 3, 2, 3}, ArmnnType); @@ -34,10 +35,8 @@ LayerTestResult<T, 4> SimpleFloorTest( 1.0f, 0.0f, 0.0f, 1.0f, 1.0f, 2.0f, 8.0f, 15.0f, 37.0f }, outputTensorInfo)); - ARMNN_NO_DEPRECATE_WARN_BEGIN - std::unique_ptr<armnn::ITensorHandle> inputHandle = workloadFactory.CreateTensorHandle(inputTensorInfo); - std::unique_ptr<armnn::ITensorHandle> outputHandle = workloadFactory.CreateTensorHandle(outputTensorInfo); - ARMNN_NO_DEPRECATE_WARN_END + std::unique_ptr<armnn::ITensorHandle> inputHandle = tensorHandleFactory.CreateTensorHandle(inputTensorInfo); + std::unique_ptr<armnn::ITensorHandle> outputHandle = tensorHandleFactory.CreateTensorHandle(outputTensorInfo); armnn::FloorQueueDescriptor data; armnn::WorkloadInfo info; @@ -65,15 +64,18 @@ LayerTestResult<T, 4> SimpleFloorTest( template LayerTestResult<armnn::ResolveType<armnn::DataType::Float32>, 4> SimpleFloorTest<armnn::DataType::Float32>( armnn::IWorkloadFactory& workloadFactory, - const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager); + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, + const armnn::ITensorHandleFactory& tensorHandleFactory); template LayerTestResult<armnn::ResolveType<armnn::DataType::Float16>, 4> SimpleFloorTest<armnn::DataType::Float16>( armnn::IWorkloadFactory& workloadFactory, - const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager); + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, + const armnn::ITensorHandleFactory& tensorHandleFactory); template LayerTestResult<armnn::ResolveType<armnn::DataType::QSymmS16>, 4> SimpleFloorTest<armnn::DataType::QSymmS16>( armnn::IWorkloadFactory& workloadFactory, - const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager); + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, + const armnn::ITensorHandleFactory& tensorHandleFactory); |