22.02
ProtoxtFixture.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_PrototxtFixture"
)
10
{
11
struct
ProtoxtTestFixture :
public
armnnUtils::ParserPrototxtFixture
<armnnOnnxParser::IOnnxParser>
12
{
13
ProtoxtTestFixture()
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
node {
24
input: "Input"
25
output: "Output"
26
name: "Plus112"
27
op_type: "Add "
28
}
29
input {
30
name: "Input"
31
type {
32
tensor_type {
33
elem_type: 1
34
shape {
35
dim {
36
dim_value: 2
37
}
38
}
39
}
40
}
41
}
42
output {
43
name: "Output"
44
type {
45
tensor_type {
46
elem_type: 1
47
shape {
48
dim {
49
dim_value: 1
50
}
51
dim {
52
dim_value: 10
53
}
54
}
55
}
56
}
57
}
58
}
59
opset_import {
60
version: 7
61
})";
62
// Setup();
63
}
64
};
65
66
67
TEST_CASE_FIXTURE
(ProtoxtTestFixture,
"ProtoxtTest"
)
68
{
69
//TODO : add a test to check if the inputs and outputs are correctly inferred.
70
}
71
72
TEST_CASE_FIXTURE
(ProtoxtTestFixture,
"ProtoxtTestWithBadInputs"
)
73
{
74
75
// CHECK_THROWS_AS(RunTest<4>({{ "InexistantInput" , {0.0, 1.0, 2.0, 3.0}}},
76
// {{ "InexistantOutput" , {0.0, 1.0, 2.0, 3.0}}}),
77
// armnn::InvalidArgumentException );
78
}
79
80
}
TEST_SUITE
TEST_SUITE("OnnxParser_PrototxtFixture")
Definition:
ProtoxtFixture.cpp:9
TEST_CASE_FIXTURE
TEST_CASE_FIXTURE(ClContextControlFixture, "CopyBetweenNeonAndGpu")
Definition:
MemCopyTests.cpp:89
ParserPrototxtFixture.hpp
IOnnxParser.hpp
armnnUtils::ParserPrototxtFixture
Definition:
ParserPrototxtFixture.hpp:24
src
armnnOnnxParser
test
ProtoxtFixture.cpp
Generated on Wed Mar 9 2022 12:00:08 for ArmNN by
1.8.13