ArmNN
 20.11
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()
BOOST_AUTO_TEST_SUITE
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
ParserPrototxtFixture.hpp
IOnnxParser.hpp
BOOST_AUTO_TEST_SUITE_END
BOOST_AUTO_TEST_SUITE_END()
BOOST_FIXTURE_TEST_CASE
BOOST_FIXTURE_TEST_CASE(ValidDepthConvTest, SimpleDepthConv2DFixture)
Definition: DepthConv.cpp:156