Gitiles
Code Review Sign In
review.mlplatform.org / ml / armnn / 246bd463ca8050c727a92dba8426a2129ba6cb53 / . / src / backends / backendsCommon / test / layerTests / StackTestImpl.hpp
blob: 90327dded7ee1f9d8366d3cc22440a9cfddd7d56 [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
Matthew Jackson9bff1442019-09-12 09:08:23 +0100[diff] [blame]10#include <Half.hpp>
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]11
Matteo Martincighe5b8eb92019-11-28 15:45:42 +0000[diff] [blame]12#include <armnn/backends/IBackendInternal.hpp>
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]13#include <backendsCommon/WorkloadFactory.hpp>
14
Matthew Jacksone69c3992019-09-09 14:31:21 +0100[diff] [blame]15LayerTestResult<float, 4> StackAxis0Float32Test(
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]16 armnn::IWorkloadFactory& workloadFactory,
Matthew Jacksone69c3992019-09-09 14:31:21 +0100[diff] [blame]17 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]18
Matthew Jacksone69c3992019-09-09 14:31:21 +0100[diff] [blame]19LayerTestResult<float, 4> StackOutput4DAxis1Float32Test(
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]20 armnn::IWorkloadFactory& workloadFactory,
Matthew Jacksone69c3992019-09-09 14:31:21 +0100[diff] [blame]21 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]22
Matthew Jacksone69c3992019-09-09 14:31:21 +0100[diff] [blame]23LayerTestResult<float, 4> StackOutput4DAxis2Float32Test(
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]24 armnn::IWorkloadFactory& workloadFactory,
Matthew Jacksone69c3992019-09-09 14:31:21 +0100[diff] [blame]25 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]26
Matthew Jacksone69c3992019-09-09 14:31:21 +0100[diff] [blame]27LayerTestResult<float, 4> StackOutput4DAxis3Float32Test(
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]28 armnn::IWorkloadFactory& workloadFactory,
Matthew Jacksone69c3992019-09-09 14:31:21 +0100[diff] [blame]29 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]30
Matthew Jacksone69c3992019-09-09 14:31:21 +0100[diff] [blame]31LayerTestResult<float, 3> StackOutput3DInputs3Float32Test(
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]32 armnn::IWorkloadFactory& workloadFactory,
Matthew Jacksone69c3992019-09-09 14:31:21 +0100[diff] [blame]33 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]34
Matthew Jacksone69c3992019-09-09 14:31:21 +0100[diff] [blame]35LayerTestResult<float, 5> StackOutput5DFloat32Test(
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]36 armnn::IWorkloadFactory& workloadFactory,
Matthew Jacksone69c3992019-09-09 14:31:21 +0100[diff] [blame]37 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]38
Matthew Jacksone69c3992019-09-09 14:31:21 +0100[diff] [blame]39LayerTestResult<armnn::Half, 4> StackFloat16Test(
Aron Virginas-Tar00d306e2019-08-28 18:08:46 +0100[diff] [blame]40 armnn::IWorkloadFactory& workloadFactory,
Matthew Jacksone69c3992019-09-09 14:31:21 +0100[diff] [blame]41 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager);