Gitiles
Code Review Sign In
review.mlplatform.org / ml / armnn / 6a1d506cb0619c6bcf8685ada56ddf4507c2f2d4 / . / src / backends / backendsCommon / test / layerTests / ArgMinMaxTestImpl.hpp
blob: b850af39f02248f448b8212e7924346f7328307a [file] [log] [blame]
Nikhil Raj68c2c902019-09-19 11:21:11 +0100[diff] [blame]1//
Sadik Armagan43528ac2020-08-27 09:06:40 +0100[diff] [blame]2// Copyright © 2019 Arm Ltd and Contributors. All rights reserved.
Nikhil Raj68c2c902019-09-19 11:21:11 +0100[diff] [blame]3// SPDX-License-Identifier: MIT
4//
5
6#pragma once
7
Sadik Armagana097d2a2021-11-24 15:47:28 +0000[diff] [blame]8#include <armnnTestUtils/LayerTestResult.hpp>
Nikhil Raj68c2c902019-09-19 11:21:11 +0100[diff] [blame]9
10#include <ResolveType.hpp>
11
Matteo Martincighe5b8eb92019-11-28 15:45:42 +0000[diff] [blame]12#include <armnn/backends/IBackendInternal.hpp>
Colm Donelan0c479742021-12-10 12:43:54 +0000[diff] [blame]13#include <armnn/backends/WorkloadFactory.hpp>
Nikhil Raj68c2c902019-09-19 11:21:11 +0100[diff] [blame]14
15template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
16LayerTestResult<int32_t, 3> ArgMaxSimpleTest(armnn::IWorkloadFactory& workloadFactory,
Sadik Armagan43528ac2020-08-27 09:06:40 +0100[diff] [blame]17 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
Sadik Armagan89de3b42020-08-28 10:38:53 +0100[diff] [blame]18 const armnn::ITensorHandleFactory& tensorHandleFactory);
Nikhil Raj68c2c902019-09-19 11:21:11 +0100[diff] [blame]19
20template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
21LayerTestResult<int32_t, 3> ArgMinSimpleTest(armnn::IWorkloadFactory& workloadFactory,
Sadik Armagan43528ac2020-08-27 09:06:40 +0100[diff] [blame]22 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
Sadik Armagan89de3b42020-08-28 10:38:53 +0100[diff] [blame]23 const armnn::ITensorHandleFactory& tensorHandleFactory);
Nikhil Raj68c2c902019-09-19 11:21:11 +0100[diff] [blame]24
25template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
James Conroyd47a0642019-09-17 14:22:06 +0100[diff] [blame]26LayerTestResult<int32_t, 3> ArgMinChannelTest(armnn::IWorkloadFactory& workloadFactory,
Sadik Armagan43528ac2020-08-27 09:06:40 +0100[diff] [blame]27 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
Sadik Armagan89de3b42020-08-28 10:38:53 +0100[diff] [blame]28 const armnn::ITensorHandleFactory& tensorHandleFactory);
Nikhil Raj68c2c902019-09-19 11:21:11 +0100[diff] [blame]29
30template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
James Conroyd47a0642019-09-17 14:22:06 +0100[diff] [blame]31LayerTestResult<int32_t, 3> ArgMaxChannelTest(armnn::IWorkloadFactory& workloadFactory,
Sadik Armagan43528ac2020-08-27 09:06:40 +0100[diff] [blame]32 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
Sadik Armagan89de3b42020-08-28 10:38:53 +0100[diff] [blame]33 const armnn::ITensorHandleFactory& tensorHandleFactory);
James Conroyd47a0642019-09-17 14:22:06 +0100[diff] [blame]34
35template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
36LayerTestResult<int32_t, 3> ArgMaxHeightTest(armnn::IWorkloadFactory& workloadFactory,
Sadik Armagan43528ac2020-08-27 09:06:40 +0100[diff] [blame]37 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
Sadik Armagan89de3b42020-08-28 10:38:53 +0100[diff] [blame]38 const armnn::ITensorHandleFactory& tensorHandleFactory);
James Conroyd47a0642019-09-17 14:22:06 +0100[diff] [blame]39
40template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
41LayerTestResult<int32_t, 3> ArgMinWidthTest(armnn::IWorkloadFactory& workloadFactory,
Sadik Armagan43528ac2020-08-27 09:06:40 +0100[diff] [blame]42 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
Sadik Armagan89de3b42020-08-28 10:38:53 +0100[diff] [blame]43 const armnn::ITensorHandleFactory& tensorHandleFactory);