22.02
Relu.cpp
Go to the documentation of this file.
1
//
2
// Copyright © 2017 Arm Ltd. All rights reserved.
3
// SPDX-License-Identifier: MIT
4
//
5
6
#include "
armnnOnnxParser/IOnnxParser.hpp
"
7
#include "
ParserPrototxtFixture.hpp
"
8
9
TEST_SUITE
(
"OnnxParser_Relu"
)
10
{
11
struct
ReluMainFixture :
public
armnnUtils::ParserPrototxtFixture
<armnnOnnxParser::IOnnxParser>
12
{
13
ReluMainFixture()
14
{
15
m_Prototext = R
"(
16
ir_version: 3
17
producer_name: "CNTK"
18
producer_version: "2.5.1"
19
domain: "ai.cntk"
20
model_version: 1
21
graph {
22
name: "CNTKGraph"
23
input {
24
name: "Input"
25
type {
26
tensor_type {
27
elem_type: 1
28
shape {
29
dim {
30
dim_value: 4
31
}
32
}
33
}
34
}
35
}
36
node {
37
input: "Input"
38
output: "Output"
39
name: "ActivationLayer"
40
op_type: "Relu"
41
}
42
output {
43
name: "Output"
44
type {
45
tensor_type {
46
elem_type: 1
47
shape {
48
dim {
49
dim_value: 4
50
}
51
}
52
}
53
}
54
}
55
}
56
opset_import {
57
version: 7
58
})";
59
Setup
();
60
}
61
};
62
63
TEST_CASE_FIXTURE
(ReluMainFixture,
"ValidReluTest"
)
64
{
65
RunTest<1>({{
"Input"
, { -1.0f, -0.5f, 1.25f, -3.0f}}},
66
{{
"Output"
, { 0.0f, 0.0f, 1.25f, 0.0f}}});
67
}
68
69
}
TEST_SUITE
TEST_SUITE("OnnxParser_Relu")
Definition:
Relu.cpp:9
TEST_CASE_FIXTURE
TEST_CASE_FIXTURE(ClContextControlFixture, "CopyBetweenNeonAndGpu")
Definition:
MemCopyTests.cpp:89
ParserPrototxtFixture.hpp
IOnnxParser.hpp
armnnUtils::ParserPrototxtFixture::Setup
void Setup()
Definition:
ParserPrototxtFixture.hpp:157
armnnUtils::ParserPrototxtFixture
Definition:
ParserPrototxtFixture.hpp:24
src
armnnOnnxParser
test
Relu.cpp
Generated on Wed Mar 9 2022 12:00:08 for ArmNN by
1.8.13