blob: 8c5385f5278c7b444d4b9bc742705a07d0afa825 [file] [log] [blame]
Aron Virginas-Tare89ebad2019-08-27 18:14:26 +01001//
2// Copyright © 2017 Arm Ltd. All rights reserved.
3// SPDX-License-Identifier: MIT
4//
5
6#pragma once
7
8#include "LayerTestResult.hpp"
9
Matteo Martincighe5b8eb92019-11-28 15:45:42 +000010#include <armnn/backends/IBackendInternal.hpp>
Aron Virginas-Tare89ebad2019-08-27 18:14:26 +010011#include <backendsCommon/WorkloadFactory.hpp>
12
13LayerTestResult<float, 4> AdditionTest(
14 armnn::IWorkloadFactory& workloadFactory,
15 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
16
17LayerTestResult<float, 5> Addition5dTest(
18 armnn::IWorkloadFactory& workloadFactory,
19 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
20
21LayerTestResult<float, 4> AdditionBroadcast1ElementTest(
22 armnn::IWorkloadFactory& workloadFactory,
23 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
24
25LayerTestResult<float, 4> AdditionBroadcastTest(
26 armnn::IWorkloadFactory& workloadFactory,
27 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
28
29LayerTestResult<uint8_t, 4> AdditionUint8Test(
30 armnn::IWorkloadFactory& workloadFactory,
31 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
32
33LayerTestResult<uint8_t, 4> AdditionBroadcast1ElementUint8Test(
34 armnn::IWorkloadFactory& workloadFactory,
35 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
36
37LayerTestResult<uint8_t, 4> AdditionBroadcastUint8Test(
38 armnn::IWorkloadFactory& workloadFactory,
39 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
40
41LayerTestResult<int16_t, 4> AdditionInt16Test(
42 armnn::IWorkloadFactory& workloadFactory,
43 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
44
45LayerTestResult<int16_t, 4> AdditionBroadcastInt16Test(
46 armnn::IWorkloadFactory& workloadFactory,
47 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
48
49LayerTestResult<int16_t, 4> AdditionBroadcast1ElementInt16Test(
50 armnn::IWorkloadFactory& workloadFactory,
51 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
52
Teresa Charlinecb6b8e2020-05-22 18:08:23 +010053LayerTestResult<int32_t, 4> AdditionInt32Test(
54 armnn::IWorkloadFactory& workloadFactory,
55 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
56
57LayerTestResult<int32_t, 4> AdditionBroadcastInt32Test(
58 armnn::IWorkloadFactory& workloadFactory,
59 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
60
61LayerTestResult<int32_t, 4> AdditionBroadcast1ElementInt32Test(
62 armnn::IWorkloadFactory& workloadFactory,
63 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
64
Aron Virginas-Tare89ebad2019-08-27 18:14:26 +010065LayerTestResult<float, 4> AdditionAfterMaxPoolTest(
66 armnn::IWorkloadFactory& workloadFactory,
67 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
68
69LayerTestResult<float, 4> CompareAdditionTest(
70 armnn::IWorkloadFactory& workloadFactory,
71 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
72 armnn::IWorkloadFactory& refWorkloadFactory);