Gitiles
Code Review Sign In
review.mlplatform.org / ml / armnn / b0b802f979a647ace8676dfdda53f06416d3700b / . / src / backends / backendsCommon / test / layerTests / StackTestImpl.hpp
blob: a2eb3a12cc6a4fd6e29dee23e88f82675e29973d [file] [log] [blame]
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]1//
Matthew Jacksone69c3992019-09-09 14:31:21 +0100[diff] [blame]2// Copyright © 2019 Arm Ltd. All rights reserved.
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]3// SPDX-License-Identifier: MIT
4//
5
6#pragma once
7
8#include "LayerTestResult.hpp"
9
10#include <ResolveType.hpp>
11
12#include <armnn/ArmNN.hpp>
13
14#include <backendsCommon/IBackendInternal.hpp>
15#include <backendsCommon/WorkloadFactory.hpp>
16
Matthew Jacksone69c3992019-09-09 14:31:21 +0100[diff] [blame]17LayerTestResult<float, 4> StackAxis0Float32Test(
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]18 armnn::IWorkloadFactory& workloadFactory,
Matthew Jacksone69c3992019-09-09 14:31:21 +0100[diff] [blame]19 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]20
Matthew Jacksone69c3992019-09-09 14:31:21 +0100[diff] [blame]21LayerTestResult<float, 4> StackOutput4DAxis1Float32Test(
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]22 armnn::IWorkloadFactory& workloadFactory,
Matthew Jacksone69c3992019-09-09 14:31:21 +0100[diff] [blame]23 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]24
Matthew Jacksone69c3992019-09-09 14:31:21 +0100[diff] [blame]25LayerTestResult<float, 4> StackOutput4DAxis2Float32Test(
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]26 armnn::IWorkloadFactory& workloadFactory,
Matthew Jacksone69c3992019-09-09 14:31:21 +0100[diff] [blame]27 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]28
Matthew Jacksone69c3992019-09-09 14:31:21 +0100[diff] [blame]29LayerTestResult<float, 4> StackOutput4DAxis3Float32Test(
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]30 armnn::IWorkloadFactory& workloadFactory,
Matthew Jacksone69c3992019-09-09 14:31:21 +0100[diff] [blame]31 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]32
Matthew Jacksone69c3992019-09-09 14:31:21 +0100[diff] [blame]33LayerTestResult<float, 3> StackOutput3DInputs3Float32Test(
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]34 armnn::IWorkloadFactory& workloadFactory,
Matthew Jacksone69c3992019-09-09 14:31:21 +0100[diff] [blame]35 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]36
Matthew Jacksone69c3992019-09-09 14:31:21 +0100[diff] [blame]37LayerTestResult<float, 5> StackOutput5DFloat32Test(
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]38 armnn::IWorkloadFactory& workloadFactory,
Matthew Jacksone69c3992019-09-09 14:31:21 +0100[diff] [blame]39 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]40
Matthew Jacksone69c3992019-09-09 14:31:21 +0100[diff] [blame]41LayerTestResult<armnn::Half, 4> StackFloat16Test(
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]42 armnn::IWorkloadFactory& workloadFactory,
Matthew Jacksone69c3992019-09-09 14:31:21 +0100[diff] [blame]43 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);