blob: 80ad54207727760cd2d0139dcc12fea65ec04ea5 [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#pragma once
6
7#include "LayerTestResult.hpp"
8
Matthew Jackson9bff1442019-09-12 09:08:23 +01009#include <Half.hpp>
10
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +010011#include <backendsCommon/IBackendInternal.hpp>
12#include <backendsCommon/WorkloadFactory.hpp>
13
14LayerTestResult<uint8_t, 4> SpaceToDepthNchwAsymmQ8Test(
15 armnn::IWorkloadFactory& workloadFactory,
16 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
17
18LayerTestResult<uint8_t, 4> SpaceToDepthNhwcAsymmQ8Test(
19 armnn::IWorkloadFactory& workloadFactory,
20 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
21
Matthew Jackson9bff1442019-09-12 09:08:23 +010022LayerTestResult<armnn::Half, 4> SpaceToDepthNchwFloat16Test(
23 armnn::IWorkloadFactory& workloadFactory,
24 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
25
26LayerTestResult<armnn::Half, 4> SpaceToDepthNhwcFloat16Test(
27 armnn::IWorkloadFactory& workloadFactory,
28 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
29
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +010030LayerTestResult<float, 4> SpaceToDepthNhwcFloat32Test1(
31 armnn::IWorkloadFactory& workloadFactory,
32 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
33
34LayerTestResult<float, 4> SpaceToDepthNchwFloat32Test1(
35 armnn::IWorkloadFactory& workloadFactory,
36 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
37
38LayerTestResult<float, 4> SpaceToDepthNhwcFloat32Test2(
39 armnn::IWorkloadFactory& workloadFactory,
40 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
41
42LayerTestResult<float, 4> SpaceToDepthNchwFloat32Test2(
43 armnn::IWorkloadFactory& workloadFactory,
44 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
45
46LayerTestResult<int16_t, 4> SpaceToDepthNhwcQSymm16Test(
47 armnn::IWorkloadFactory& workloadFactory,
48 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
49
50LayerTestResult<int16_t, 4> SpaceToDepthNchwQSymm16Test(
51 armnn::IWorkloadFactory& workloadFactory,
52 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);