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