Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 1 | // |
| 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 Martincigh | e5b8eb9 | 2019-11-28 15:45:42 +0000 | [diff] [blame] | 10 | #include <armnn/backends/IBackendInternal.hpp> |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 11 | #include <backendsCommon/WorkloadFactory.hpp> |
| 12 | |
| 13 | // |
| 14 | // Sigmoid |
| 15 | // |
| 16 | |
| 17 | LayerTestResult<float, 4> SimpleSigmoidTest( |
| 18 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 19 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 20 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 21 | |
| 22 | LayerTestResult<uint8_t, 4> SimpleSigmoidUint8Test( |
| 23 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 24 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 25 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 26 | |
| 27 | LayerTestResult<int16_t, 4> SimpleSigmoidInt16Test( |
| 28 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 29 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 30 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 31 | |
| 32 | // |
| 33 | // TanH |
| 34 | // |
| 35 | |
| 36 | LayerTestResult<float, 4> TanhTest( |
| 37 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 38 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 39 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 40 | |
| 41 | LayerTestResult<uint8_t, 4> TanhUint8Test( |
| 42 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 43 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 44 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 45 | |
| 46 | LayerTestResult<int16_t, 4> TanhInt16Test( |
| 47 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 48 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 49 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 50 | |
| 51 | // |
| 52 | // Linear |
| 53 | // |
| 54 | |
| 55 | LayerTestResult<float, 4> ConstantLinearActivationTest( |
| 56 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 57 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 58 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 59 | |
| 60 | LayerTestResult<uint8_t, 4> ConstantLinearActivationUint8Test( |
| 61 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 62 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 63 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 64 | |
| 65 | LayerTestResult<int16_t, 4> ConstantLinearActivationInt16Test( |
| 66 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 67 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 68 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 69 | |
| 70 | // |
| 71 | // ReLu |
| 72 | // |
| 73 | |
| 74 | LayerTestResult<float, 4> ReLuTest( |
| 75 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 76 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 77 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 78 | |
| 79 | LayerTestResult<uint8_t, 4> ReLuUint8Test( |
| 80 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 81 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 82 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 83 | |
| 84 | LayerTestResult<int16_t, 4> ReLuInt16Test( |
| 85 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 86 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 87 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 88 | |
| 89 | // |
| 90 | // BoundedReLu |
| 91 | // |
| 92 | |
| 93 | LayerTestResult<uint8_t, 4> BoundedReLuUint8Test( |
| 94 | armnn::IWorkloadFactory& workloadFactory, |
| 95 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 96 | float upperBound); |
| 97 | |
| 98 | LayerTestResult<uint8_t, 4> BoundedReLuUint8Test( |
| 99 | armnn::IWorkloadFactory& workloadFactory, |
| 100 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 101 | float upperBound, |
| 102 | float lowerBound); |
| 103 | |
| 104 | LayerTestResult<int16_t, 4> BoundedReLuInt16Test( |
| 105 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 106 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 107 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 108 | |
| 109 | LayerTestResult<float, 4> BoundedReLuUpperAndLowerBoundTest( |
| 110 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 111 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 112 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 113 | |
| 114 | LayerTestResult<uint8_t, 4> BoundedReLuUint8UpperAndLowerBoundTest( |
| 115 | armnn::IWorkloadFactory& workloadFactor, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 116 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManagery, |
| 117 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 118 | |
| 119 | LayerTestResult<float, 4> BoundedReLuUpperBoundOnlyTest( |
| 120 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 121 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 122 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 123 | |
| 124 | LayerTestResult<uint8_t, 4> BoundedReLuUint8UpperBoundOnlyTest( |
| 125 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 126 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 127 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 128 | |
| 129 | LayerTestResult<float, 4> CompareBoundedReLuTest( |
| 130 | armnn::IWorkloadFactory& workloadFactory, |
| 131 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 132 | armnn::IWorkloadFactory& refWorkloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 133 | const armnn::ITensorHandleFactory& tensorHandleFactory, |
| 134 | const armnn::ITensorHandleFactory& refTensorHandleFactory, |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 135 | float upperBound, |
| 136 | float lowerBound); |
| 137 | |
| 138 | // |
| 139 | // SoftReLu |
| 140 | // |
| 141 | |
| 142 | LayerTestResult<float, 4> SoftReLuTest( |
| 143 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 144 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 145 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 146 | |
| 147 | LayerTestResult<uint8_t, 4> SoftReLuUint8Test( |
| 148 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 149 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 150 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 151 | |
| 152 | LayerTestResult<int16_t, 4> SoftReLuInt16Test( |
| 153 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 154 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 155 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 156 | |
| 157 | // |
| 158 | // LeakyReLu |
| 159 | // |
| 160 | |
| 161 | LayerTestResult<float, 4> LeakyReLuTest( |
| 162 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 163 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 164 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 165 | |
| 166 | LayerTestResult<uint8_t, 4> LeakyReLuUint8Test( |
| 167 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 168 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 169 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 170 | |
| 171 | LayerTestResult<int16_t, 4> LeakyReLuInt16Test( |
| 172 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 173 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 174 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 175 | |
| 176 | // |
David Monahan | 3b3c381 | 2020-02-25 09:03:29 +0000 | [diff] [blame] | 177 | // Elu |
| 178 | // |
| 179 | |
| 180 | LayerTestResult<float, 4> EluTest( |
| 181 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 182 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 183 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
David Monahan | 3b3c381 | 2020-02-25 09:03:29 +0000 | [diff] [blame] | 184 | |
| 185 | LayerTestResult<uint8_t, 4> EluUint8Test( |
| 186 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 187 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 188 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
David Monahan | 3b3c381 | 2020-02-25 09:03:29 +0000 | [diff] [blame] | 189 | |
| 190 | LayerTestResult<int16_t, 4> EluInt16Test( |
| 191 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 192 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 193 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
David Monahan | 3b3c381 | 2020-02-25 09:03:29 +0000 | [diff] [blame] | 194 | |
| 195 | // |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 196 | // Abs |
| 197 | // |
| 198 | |
| 199 | LayerTestResult<float, 4> AbsTest( |
| 200 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 201 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 202 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 203 | |
| 204 | LayerTestResult<uint8_t, 4> AbsUint8Test( |
| 205 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 206 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 207 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 208 | |
| 209 | LayerTestResult<int16_t, 4> AbsInt16Test( |
| 210 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 211 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 212 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 213 | |
| 214 | // |
| 215 | // Sqrt |
| 216 | // |
| 217 | |
| 218 | LayerTestResult<float, 4> SqrtTest( |
| 219 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 220 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 221 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 222 | |
Sadik Armagan | 6095ba5 | 2019-09-13 17:07:19 +0100 | [diff] [blame] | 223 | LayerTestResult<float, 5> SqrtNNTest( |
| 224 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 225 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 226 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Sadik Armagan | 6095ba5 | 2019-09-13 17:07:19 +0100 | [diff] [blame] | 227 | |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 228 | LayerTestResult<uint8_t, 4> SqrtUint8Test( |
| 229 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 230 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 231 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 232 | |
| 233 | LayerTestResult<int16_t, 4> SqrtInt16Test( |
| 234 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 235 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 236 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 237 | |
| 238 | // |
| 239 | // Square |
| 240 | // |
| 241 | |
| 242 | LayerTestResult<float, 4> SquareTest( |
| 243 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 244 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 245 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 246 | |
| 247 | LayerTestResult<uint8_t, 4> SquareUint8Test( |
| 248 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 249 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 250 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 251 | |
| 252 | LayerTestResult<int16_t, 4> SquareInt16Test( |
| 253 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 254 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 255 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 256 | |
| 257 | // |
Colm Donelan | 03fbeaf | 2020-02-26 15:39:23 +0000 | [diff] [blame] | 258 | // HardSwish |
| 259 | // |
| 260 | |
| 261 | LayerTestResult<float, 4> HardSwishTest( |
| 262 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 263 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 264 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Colm Donelan | 03fbeaf | 2020-02-26 15:39:23 +0000 | [diff] [blame] | 265 | |
| 266 | LayerTestResult<uint8_t, 4> HardSwishUint8Test( |
| 267 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 268 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 269 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Colm Donelan | 03fbeaf | 2020-02-26 15:39:23 +0000 | [diff] [blame] | 270 | |
| 271 | LayerTestResult<int16_t, 4> HardSwishInt16Test( |
| 272 | armnn::IWorkloadFactory& workloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 273 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 274 | const armnn::ITensorHandleFactory& tensorHandleFactory); |
Colm Donelan | 03fbeaf | 2020-02-26 15:39:23 +0000 | [diff] [blame] | 275 | |
| 276 | // |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 277 | // Other |
| 278 | // |
| 279 | |
| 280 | LayerTestResult<float, 4> CompareActivationTest( |
| 281 | armnn::IWorkloadFactory& workloadFactory, |
| 282 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 283 | armnn::IWorkloadFactory& refWorkloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 284 | const armnn::ITensorHandleFactory& tensorHandleFactory, |
| 285 | const armnn::ITensorHandleFactory& refTensorHandleFactory, |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 286 | armnn::ActivationFunction f, |
| 287 | unsigned int batchSize); |
| 288 | |
| 289 | LayerTestResult<uint8_t, 4> CompareActivationUint8Test( |
| 290 | armnn::IWorkloadFactory& workloadFactory, |
| 291 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 292 | armnn::IWorkloadFactory& refWorkloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 293 | const armnn::ITensorHandleFactory& tensorHandleFactory, |
| 294 | const armnn::ITensorHandleFactory& refTensorHandleFactory, |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 295 | armnn::ActivationFunction f); |
| 296 | |
| 297 | LayerTestResult<int16_t, 4> CompareActivationInt16Test( |
| 298 | armnn::IWorkloadFactory& workloadFactory, |
| 299 | const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, |
| 300 | armnn::IWorkloadFactory& refWorkloadFactory, |
Keith Davis | 33a626f | 2020-08-27 15:38:12 +0100 | [diff] [blame] | 301 | const armnn::ITensorHandleFactory& tensorHandleFactory, |
| 302 | const armnn::ITensorHandleFactory& refTensorHandleFactory, |
Aron Virginas-Tar | 00d306e | 2019-08-28 18:08:46 +0100 | [diff] [blame] | 303 | armnn::ActivationFunction f); |