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/SoftmaxTestImpl.hpp | |
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/SoftmaxTestImpl.hpp')
-rw-r--r-- | src/backends/backendsCommon/test/layerTests/SoftmaxTestImpl.hpp | 38 |
1 files changed, 19 insertions, 19 deletions
diff --git a/src/backends/backendsCommon/test/layerTests/SoftmaxTestImpl.hpp b/src/backends/backendsCommon/test/layerTests/SoftmaxTestImpl.hpp index 478d3803a1..e95cd5e8cd 100644 --- a/src/backends/backendsCommon/test/layerTests/SoftmaxTestImpl.hpp +++ b/src/backends/backendsCommon/test/layerTests/SoftmaxTestImpl.hpp @@ -16,108 +16,108 @@ LayerTestResult<float, 2> SimpleSoftmaxTest( armnn::IWorkloadFactory& workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, - armnn::ITensorHandleFactory* tensorHandleFactory, + const armnn::ITensorHandleFactory& tensorHandleFactory, float beta); LayerTestResult<float, 2> SimpleAxisSoftmaxTest( armnn::IWorkloadFactory& workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, - armnn::ITensorHandleFactory* tensorHandleFactory, + const armnn::ITensorHandleFactory& tensorHandleFactory, float beta, int axis); LayerTestResult<float, 3> Simple3dSoftmaxTest( armnn::IWorkloadFactory& workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, - armnn::ITensorHandleFactory* tensorHandleFactory, + const armnn::ITensorHandleFactory& tensorHandleFactory, float beta); LayerTestResult<float, 3> Simple3dAxisSoftmaxTest( armnn::IWorkloadFactory& workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, - armnn::ITensorHandleFactory* tensorHandleFactory, + const armnn::ITensorHandleFactory& tensorHandleFactory, float beta, int axis); LayerTestResult<float, 4> Simple4dSoftmaxTest( armnn::IWorkloadFactory& workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, - armnn::ITensorHandleFactory* tensorHandleFactory, + const armnn::ITensorHandleFactory& tensorHandleFactory, float beta); LayerTestResult<float, 4> Simple4dAxisSoftmaxTest( armnn::IWorkloadFactory& workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, - armnn::ITensorHandleFactory* tensorHandleFactory, + const armnn::ITensorHandleFactory& tensorHandleFactory, float beta, int axis); LayerTestResult<uint8_t, 2> SimpleSoftmaxUint8Test( armnn::IWorkloadFactory& workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, - armnn::ITensorHandleFactory* tensorHandleFactory, + const armnn::ITensorHandleFactory& tensorHandleFactory, float beta); LayerTestResult<uint8_t,3> Simple3dSoftmaxUint8Test( armnn::IWorkloadFactory& workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, - armnn::ITensorHandleFactory* tensorHandleFactory, + const armnn::ITensorHandleFactory& tensorHandleFactory, float beta); LayerTestResult<uint8_t,4> Simple4dSoftmaxUint8Test( armnn::IWorkloadFactory& workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, - armnn::ITensorHandleFactory* tensorHandleFactory, + const armnn::ITensorHandleFactory& tensorHandleFactory, float beta); LayerTestResult<armnn::Half,2> SimpleSoftmaxFloat16Test( armnn::IWorkloadFactory& workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, - armnn::ITensorHandleFactory* tensorHandleFactory, + const armnn::ITensorHandleFactory& tensorHandleFactory, float beta); LayerTestResult<armnn::Half,3> Simple3dSoftmaxFloat16Test( armnn::IWorkloadFactory& workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, - armnn::ITensorHandleFactory* tensorHandleFactory, + const armnn::ITensorHandleFactory& tensorHandleFactory, float beta); LayerTestResult<armnn::Half,4> Simple4dSoftmaxFloat16Test( armnn::IWorkloadFactory& workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, - armnn::ITensorHandleFactory* tensorHandleFactory, + const armnn::ITensorHandleFactory& tensorHandleFactory, float beta); LayerTestResult<int16_t,2> SimpleSoftmaxUint16Test( armnn::IWorkloadFactory& workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, - armnn::ITensorHandleFactory* tensorHandleFactory, + const armnn::ITensorHandleFactory& tensorHandleFactory, float beta); LayerTestResult<int16_t,3> Simple3dSoftmaxUint16Test( armnn::IWorkloadFactory& workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, - armnn::ITensorHandleFactory* tensorHandleFactory, + const armnn::ITensorHandleFactory& tensorHandleFactory, float beta); LayerTestResult<int16_t,4> Simple4dSoftmaxUint16Test( armnn::IWorkloadFactory& workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, - armnn::ITensorHandleFactory* tensorHandleFactory, + const armnn::ITensorHandleFactory& tensorHandleFactory, float beta); LayerTestResult<float, 2> CompareSoftmaxTest( armnn::IWorkloadFactory& workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, armnn::IWorkloadFactory& refWorkloadFactory, - const armnn::ITensorHandleFactory* tensorHandleFactory, - const armnn::ITensorHandleFactory* refTensorHandleFactory, + const armnn::ITensorHandleFactory& tensorHandleFactory, + const armnn::ITensorHandleFactory& refTensorHandleFactory, float beta); LayerTestResult<uint8_t, 2> CompareSoftmaxUint8Test( armnn::IWorkloadFactory& workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, armnn::IWorkloadFactory& refWorkloadFactory, - const armnn::ITensorHandleFactory* tensorHandleFactory, - const armnn::ITensorHandleFactory* refTensorHandleFactory, + const armnn::ITensorHandleFactory& tensorHandleFactory, + const armnn::ITensorHandleFactory& refTensorHandleFactory, float beta); |