diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2020-08-28 10:38:53 +0100 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2020-08-28 10:38:53 +0100 |
commit | 89de3b43a558710c7b1db2f4409b515d2fd94f0b (patch) | |
tree | b8c13b9ec9be56b3cef60e661885e09b5e9edadc /src/backends/backendsCommon/test/layerTests/FillTestImpl.cpp | |
parent | 56785c75037ed0cd377851616634b3129713394b (diff) | |
download | armnn-89de3b43a558710c7b1db2f4409b515d2fd94f0b.tar.gz |
IVGCVSW-4979 'Remove CreateTensorHandle using WorkloadFactory in workload tests'
* Small refactor in unit tests using TensorHandleFactory to use reference instead of pointer
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I1a702941890034a45029c014c8b11e185f45a807
Diffstat (limited to 'src/backends/backendsCommon/test/layerTests/FillTestImpl.cpp')
-rw-r--r-- | src/backends/backendsCommon/test/layerTests/FillTestImpl.cpp | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/src/backends/backendsCommon/test/layerTests/FillTestImpl.cpp b/src/backends/backendsCommon/test/layerTests/FillTestImpl.cpp index f848cfd905..f96d33735f 100644 --- a/src/backends/backendsCommon/test/layerTests/FillTestImpl.cpp +++ b/src/backends/backendsCommon/test/layerTests/FillTestImpl.cpp @@ -15,7 +15,7 @@ template<armnn::DataType ArmnnType, typename T> LayerTestResult<T, 4> SimpleFillTest( armnn::IWorkloadFactory& workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, - armnn::ITensorHandleFactory* tensorHandleFactory) + const armnn::ITensorHandleFactory& tensorHandleFactory) { IgnoreUnused(memoryManager); armnn::TensorInfo inputTensorInfo({4}, armnn::DataType::Signed32); @@ -31,8 +31,8 @@ LayerTestResult<T, 4> SimpleFillTest( 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f }, outputTensorInfo)); - std::unique_ptr<armnn::ITensorHandle> inputHandle = tensorHandleFactory->CreateTensorHandle(inputTensorInfo); - std::unique_ptr<armnn::ITensorHandle> outputHandle = tensorHandleFactory->CreateTensorHandle(outputTensorInfo); + std::unique_ptr<armnn::ITensorHandle> inputHandle = tensorHandleFactory.CreateTensorHandle(inputTensorInfo); + std::unique_ptr<armnn::ITensorHandle> outputHandle = tensorHandleFactory.CreateTensorHandle(outputTensorInfo); armnn::FillQueueDescriptor data; data.m_Parameters.m_Value = 1.0f; @@ -61,16 +61,16 @@ template LayerTestResult<armnn::ResolveType<armnn::DataType::Float32>, 4> SimpleFillTest<armnn::DataType::Float32>( armnn::IWorkloadFactory& workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, - armnn::ITensorHandleFactory* tensorHandleFactory); + const armnn::ITensorHandleFactory& tensorHandleFactory); template LayerTestResult<armnn::ResolveType<armnn::DataType::Float16>, 4> SimpleFillTest<armnn::DataType::Float16>( armnn::IWorkloadFactory& workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, - armnn::ITensorHandleFactory* tensorHandleFactory); + const armnn::ITensorHandleFactory& tensorHandleFactory); template LayerTestResult<armnn::ResolveType<armnn::DataType::Signed32>, 4> SimpleFillTest<armnn::DataType::Signed32>( armnn::IWorkloadFactory& workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, - armnn::ITensorHandleFactory* tensorHandleFactory);
\ No newline at end of file + const armnn::ITensorHandleFactory& tensorHandleFactory);
\ No newline at end of file |