aboutsummaryrefslogtreecommitdiff
path: root/src/backends/backendsCommon/test/layerTests/SoftmaxTestImpl.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/backendsCommon/test/layerTests/SoftmaxTestImpl.hpp')
-rw-r--r--src/backends/backendsCommon/test/layerTests/SoftmaxTestImpl.hpp38
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);