blob: 00b5f7d0a70877e9ff1d2a919c9f6ffdfa8d5fe4 [file] [log] [blame]
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +01001//
2// Copyright © 2017 Arm Ltd. All rights reserved.
3// SPDX-License-Identifier: MIT
4//
5
6#include "LayerTestResult.hpp"
7
Matthew Jackson9bff1442019-09-12 09:08:23 +01008#include <Half.hpp>
9
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +010010#include <backendsCommon/IBackendInternal.hpp>
11#include <backendsCommon/WorkloadFactory.hpp>
12
13#include <vector>
14
Matthew Jackson9bff1442019-09-12 09:08:23 +010015std::vector<LayerTestResult<float, 3>> SplitterFloat32Test(
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +010016 armnn::IWorkloadFactory& workloadFactory,
17 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
18
Matthew Jackson9bff1442019-09-12 09:08:23 +010019LayerTestResult<float, 3> CopyViaSplitterFloat32Test(
20 armnn::IWorkloadFactory& workloadFactory,
21 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
22
23std::vector<LayerTestResult<armnn::Half, 3>> SplitterFloat16Test(
24 armnn::IWorkloadFactory& workloadFactory,
25 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
26
27LayerTestResult<armnn::Half, 3> CopyViaSplitterFloat16Test(
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +010028 armnn::IWorkloadFactory& workloadFactory,
29 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
30
31std::vector<LayerTestResult<uint8_t, 3>> SplitterUint8Test(
32 armnn::IWorkloadFactory& workloadFactory,
33 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
34
35std::vector<LayerTestResult<int16_t, 3>> SplitterInt16Test(
36 armnn::IWorkloadFactory& workloadFactory,
37 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
38
39LayerTestResult<uint8_t, 3> CopyViaSplitterUint8Test(
40 armnn::IWorkloadFactory& workloadFactory,
41 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
42
43LayerTestResult<int16_t, 3> CopyViaSplitterInt16Test(
44 armnn::IWorkloadFactory& workloadFactory,
45 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);