21.11
ActivationFixture.hpp
Go to the documentation of this file.
1
//
2
// Copyright © 2017 Arm Ltd. All rights reserved.
3
// SPDX-License-Identifier: MIT
4
//
5
#pragma once
6
7
#include "
TensorCopyUtils.hpp
"
8
#include "
WorkloadTestUtils.hpp
"
9
10
#include <
armnn/utility/NumericCast.hpp
>
11
12
#include <
test/TensorHelpers.hpp
>
13
14
struct
ActivationFixture
15
{
16
ActivationFixture
()
17
{
18
output
.resize(
batchSize
*
channels
*
height
*
width
);
19
outputExpected
.resize(
batchSize
*
channels
*
height
* width);
20
input
.resize(
batchSize
*
channels
*
height
* width);
21
22
unsigned
int
inputShape[] = {
batchSize
,
channels
,
height
, width };
23
unsigned
int
outputShape[] = {
batchSize
,
channels
,
height
, width };
24
25
inputTensorInfo
=
armnn::TensorInfo
(4, inputShape,
armnn::DataType::Float32
);
26
outputTensorInfo
=
armnn::TensorInfo
(4, outputShape,
armnn::DataType::Float32
);
27
28
input
= MakeRandomTensor<float>(
inputTensorInfo
, 21453);
29
}
30
31
unsigned
int
width
= 17;
32
unsigned
int
height
= 29;
33
unsigned
int
channels
= 2;
34
unsigned
int
batchSize
= 5;
35
36
std::vector<float>
output
;
37
std::vector<float>
outputExpected
;
38
std::vector<float>
input
;
39
40
armnn::TensorInfo
inputTensorInfo
;
41
armnn::TensorInfo
outputTensorInfo
;
42
43
// Parameters used by some of the activation functions.
44
float
a
= 0.234f;
45
float
b
= -12.345f;
46
};
47
48
49
struct
PositiveActivationFixture
:
public
ActivationFixture
50
{
51
PositiveActivationFixture
()
52
{
53
input
= MakeRandomTensor<float>(
inputTensorInfo
, 2342423, 0.0f, 1.0f);
54
}
55
};
PositiveActivationFixture
Definition:
ActivationFixture.hpp:49
ActivationFixture::width
unsigned int width
Definition:
ActivationFixture.hpp:31
ActivationFixture::b
float b
Definition:
ActivationFixture.hpp:45
TensorCopyUtils.hpp
ActivationFixture::input
std::vector< float > input
Definition:
ActivationFixture.hpp:38
ActivationFixture::a
float a
Definition:
ActivationFixture.hpp:44
armnn::TensorInfo
Definition:
Tensor.hpp:152
WorkloadTestUtils.hpp
ActivationFixture::channels
unsigned int channels
Definition:
ActivationFixture.hpp:33
NumericCast.hpp
ActivationFixture::outputTensorInfo
armnn::TensorInfo outputTensorInfo
Definition:
ActivationFixture.hpp:41
TensorHelpers.hpp
PositiveActivationFixture::PositiveActivationFixture
PositiveActivationFixture()
Definition:
ActivationFixture.hpp:51
ActivationFixture::ActivationFixture
ActivationFixture()
Definition:
ActivationFixture.hpp:16
ActivationFixture::inputTensorInfo
armnn::TensorInfo inputTensorInfo
Definition:
ActivationFixture.hpp:40
ActivationFixture::output
std::vector< float > output
Definition:
ActivationFixture.hpp:36
ActivationFixture
Definition:
ActivationFixture.hpp:14
ActivationFixture::outputExpected
std::vector< float > outputExpected
Definition:
ActivationFixture.hpp:37
ActivationFixture::batchSize
unsigned int batchSize
Definition:
ActivationFixture.hpp:34
armnn::DataType::Float32
ActivationFixture::height
unsigned int height
Definition:
ActivationFixture.hpp:32
src
backends
backendsCommon
test
ActivationFixture.hpp
Generated on Wed Nov 17 2021 12:59:33 for ArmNN by
1.8.13