20.11
Softmax.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 "
armnnTfParser/ITfParser.hpp
"
8
#include "
ParserPrototxtFixture.hpp
"
9
10
BOOST_AUTO_TEST_SUITE
(TensorflowParser)
11
12
struct
SoftmaxFixture :
public
armnnUtils::ParserPrototxtFixture
<armnnTfParser::ITfParser>
13
{
14
SoftmaxFixture()
15
{
16
m_Prototext
=
"node {\n"
17
" name: \"blah\"\n"
18
" op: \"Placeholder\"\n"
19
" attr {\n"
20
" key: \"dtype\"\n"
21
" value {\n"
22
" type: DT_FLOAT\n"
23
" }\n"
24
" }\n"
25
" attr {\n"
26
" key: \"shape\"\n"
27
" value {\n"
28
" shape {\n"
29
" }\n"
30
" }\n"
31
" }\n"
32
"}\n"
33
"node {\n"
34
" name: \"blah2\"\n"
35
" op: \"Softmax\"\n"
36
" input: \"blah\"\n"
37
" attr {\n"
38
" key: \"T\"\n"
39
" value {\n"
40
" type: DT_FLOAT\n"
41
" }\n"
42
" }\n"
43
"}\n"
;
44
45
SetupSingleInputSingleOutput
({ 1, 7 },
"blah"
,
"blah2"
);
46
}
47
};
48
49
BOOST_FIXTURE_TEST_CASE
(ParseSoftmax, SoftmaxFixture)
50
{
51
RunTest<2>({ 0, 0, 10000, 0, 0, 0, 0 }, { 0, 0, 1, 0, 0, 0, 0 });
52
}
53
54
55
BOOST_AUTO_TEST_SUITE_END
()
BOOST_AUTO_TEST_SUITE
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
armnnUtils::ParserPrototxtFixture< armnnTfParser::ITfParser >::m_Prototext
std::string m_Prototext
Definition:
ParserPrototxtFixture.hpp:67
ITfParser.hpp
BOOST_FIXTURE_TEST_CASE
BOOST_FIXTURE_TEST_CASE(ParseSoftmaxLite, SoftmaxFixture)
Definition:
Softmax.cpp:72
ParserPrototxtFixture.hpp
BOOST_AUTO_TEST_SUITE_END
BOOST_AUTO_TEST_SUITE_END()
armnnUtils::ParserPrototxtFixture::SetupSingleInputSingleOutput
void SetupSingleInputSingleOutput(const std::string &inputName, const std::string &outputName)
Parses and loads the network defined by the m_Prototext string.
Definition:
ParserPrototxtFixture.hpp:85
armnnUtils::ParserPrototxtFixture
Definition:
ParserPrototxtFixture.hpp:23
src
armnnTfParser
test
Softmax.cpp
Generated on Wed Nov 25 2020 09:59:11 for ArmNN by
1.8.13