blob: 44dd9cea9dd2d3a64c0aee1e1d4f7d882093d346 [file] [log] [blame]
Sadik Armagan9199e582019-09-05 17:35:31 +01001//
2// Copyright © 2017 Arm Ltd. All rights reserved.
3// SPDX-License-Identifier: MIT
4//
5
6#pragma once
7
8#include "LayerTestResult.hpp"
9
10#include <ResolveType.hpp>
11
Matteo Martincighe5b8eb92019-11-28 15:45:42 +000012#include <armnn/backends/IBackendInternal.hpp>
Sadik Armagan9199e582019-09-05 17:35:31 +010013#include <backendsCommon/WorkloadFactory.hpp>
14
15template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
16LayerTestResult<T, 2> Abs2dTest(
17 armnn::IWorkloadFactory& workloadFactory,
Keith Davis33a626f2020-08-27 15:38:12 +010018 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
19 const armnn::ITensorHandleFactory& tensorHandleFactory);
Sadik Armagan9199e582019-09-05 17:35:31 +010020
21template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
22LayerTestResult<T, 3> Abs3dTest(
23 armnn::IWorkloadFactory& workloadFactory,
Keith Davis33a626f2020-08-27 15:38:12 +010024 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
25 const armnn::ITensorHandleFactory& tensorHandleFactory);
Sadik Armagan9199e582019-09-05 17:35:31 +010026
27template<armnn::DataType ArmnnType, typename T = armnn::ResolveType<ArmnnType>>
28LayerTestResult<T, 2> AbsZeroTest(
29 armnn::IWorkloadFactory& workloadFactory,
Keith Davis33a626f2020-08-27 15:38:12 +010030 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
31 const armnn::ITensorHandleFactory& tensorHandleFactory);