ArmNN
NotReleased
DepthConv.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 <boost/test/unit_test.hpp>
7
#include "
armnnOnnxParser/IOnnxParser.hpp
"
8
#include "
ParserPrototxtFixture.hpp
"
9
10
BOOST_AUTO_TEST_SUITE
(OnnxParser)
11
12
struct
SimpleDepthConv2DFixture :
public
armnnUtils::ParserPrototxtFixture
<armnnOnnxParser::IOnnxParser>
13
{
14
SimpleDepthConv2DFixture()
15
{
16
m_Prototext
= R
"(
17
ir_version: 3
18
producer_name: "CNTK"
19
producer_version: "2.5.1"
20
domain: "ai.cntk"
21
model_version: 1
22
graph {
23
name: "CNTKGraph"
24
input {
25
name: "Input"
26
type {
27
tensor_type {
28
elem_type: 1
29
shape {
30
dim {
31
dim_value: 1
32
}
33
dim {
34
dim_value: 3
35
}
36
dim {
37
dim_value: 2
38
}
39
dim {
40
dim_value: 2
41
}
42
}
43
}
44
}
45
}
46
input {
47
name: "Weight"
48
type {
49
tensor_type {
50
elem_type: 1
51
shape {
52
dim {
53
dim_value: 3
54
}
55
dim {
56
dim_value: 1
57
}
58
dim {
59
dim_value: 2
60
}
61
dim {
62
dim_value: 2
63
}
64
}
65
}
66
}
67
}
68
initializer {
69
dims: 3
70
dims: 1
71
dims: 2
72
dims: 2
73
data_type: 1
74
float_data: 1
75
float_data: 1
76
float_data: 1
77
float_data: 1
78
float_data: 2
79
float_data: 2
80
float_data: 2
81
float_data: 2
82
float_data: 3
83
float_data: 3
84
float_data: 3
85
float_data: 3
86
name: "Weight"
87
}
88
node {
89
input: "Input"
90
input: "Weight"
91
output: "Output"
92
name: "Convolution"
93
op_type: "Conv"
94
attribute {
95
name: "kernel_shape"
96
ints: 2
97
ints: 2
98
type: INTS
99
}
100
attribute {
101
name: "strides"
102
ints: 1
103
ints: 1
104
type: INTS
105
}
106
attribute {
107
name: "auto_pad"
108
s: "VALID"
109
type: STRING
110
}
111
attribute {
112
name: "group"
113
i: 3
114
type: INT
115
}
116
attribute {
117
name: "dilations"
118
ints: 1
119
ints: 1
120
type: INTS
121
}
122
doc_string: ""
123
domain: ""
124
}
125
output {
126
name: "Output"
127
type {
128
tensor_type {
129
elem_type: 1
130
shape {
131
dim {
132
dim_value: 1
133
}
134
dim {
135
dim_value: 3
136
}
137
dim {
138
dim_value: 1
139
}
140
dim {
141
dim_value: 1
142
}
143
}
144
}
145
}
146
}
147
}
148
opset_import {
149
version: 7
150
})";
151
Setup
();
152
}
153
};
154
155
156
BOOST_FIXTURE_TEST_CASE
(ValidDepthConvTest, SimpleDepthConv2DFixture)
157
{
158
RunTest<4>({{
"Input"
, { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12}}},
159
{{
"Output"
, { 10, 52, 126 }}});
160
}
161
162
BOOST_AUTO_TEST_SUITE_END
()
IOnnxParser.hpp
armnnUtils::ParserPrototxtFixture< armnnOnnxParser::IOnnxParser >::m_Prototext
std::string m_Prototext
Definition:
ParserPrototxtFixture.hpp:68
BOOST_AUTO_TEST_SUITE_END
BOOST_AUTO_TEST_SUITE_END()
armnnUtils::ParserPrototxtFixture
Definition:
ParserPrototxtFixture.hpp:24
BOOST_AUTO_TEST_SUITE
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
ParserPrototxtFixture.hpp
BOOST_FIXTURE_TEST_CASE
BOOST_FIXTURE_TEST_CASE(ValidDepthConvTest, SimpleDepthConv2DFixture)
Definition:
DepthConv.cpp:156
armnnUtils::ParserPrototxtFixture::Setup
void Setup()
Definition:
ParserPrototxtFixture.hpp:140
src
armnnOnnxParser
test
DepthConv.cpp
Generated on Fri Mar 13 2020 16:06:55 for ArmNN by
1.8.13