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 "
ParserFlatbuffersFixture.hpp
"
8
#include "../TfLiteParser.hpp"
9
10
#include <string>
11
#include <iostream>
12
13
BOOST_AUTO_TEST_SUITE
(TensorflowLiteParser)
14
15
struct
SoftmaxFixture :
public
ParserFlatbuffersFixture
16
{
17
explicit
SoftmaxFixture()
18
{
19
m_JsonString
= R
"(
20
{
21
"version": 3,
22
"operator_codes": [ { "builtin_code": "SOFTMAX" } ],
23
"subgraphs": [ {
24
"tensors": [
25
{
26
"shape": [ 1, 7 ],
27
"type": "UINT8",
28
"buffer": 0,
29
"name": "inputTensor",
30
"quantization": {
31
"min": [ 0.0 ],
32
"max": [ 255.0 ],
33
"scale": [ 1.0 ],
34
"zero_point": [ 0 ],
35
}
36
},
37
{
38
"shape": [ 1, 7 ],
39
"type": "UINT8",
40
"buffer": 1,
41
"name": "outputTensor",
42
"quantization": {
43
"min": [ 0.0 ],
44
"max": [ 255.0 ],
45
"scale": [ 0.00390625 ],
46
"zero_point": [ 0 ],
47
}
48
}
49
],
50
"inputs": [ 0 ],
51
"outputs": [ 1 ],
52
"operators": [
53
{
54
"opcode_index": 0,
55
"inputs": [ 0 ],
56
"outputs": [ 1 ],
57
"builtin_options_type": "SoftmaxOptions",
58
"builtin_options": {
59
"beta": 1.0
60
},
61
"custom_options_format": "FLEXBUFFERS"
62
}
63
],
64
} ],
65
"buffers" : [ {}, {} ]
66
}
67
)";
68
SetupSingleInputSingleOutput
(
"inputTensor"
,
"outputTensor"
);
69
}
70
};
71
72
BOOST_FIXTURE_TEST_CASE
(ParseSoftmaxLite, SoftmaxFixture)
73
{
74
RunTest<2, armnn::DataType::QAsymmU8>(0, { 0, 0, 100, 0, 0, 0, 0 }, { 0, 0, 255, 0, 0, 0, 0 });
75
}
76
77
BOOST_AUTO_TEST_SUITE_END
()
BOOST_AUTO_TEST_SUITE
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
ParserFlatbuffersFixture::m_JsonString
std::string m_JsonString
Definition:
ParserFlatbuffersFixture.hpp:51
BOOST_FIXTURE_TEST_CASE
BOOST_FIXTURE_TEST_CASE(ParseSoftmaxLite, SoftmaxFixture)
Definition:
Softmax.cpp:72
ParserFlatbuffersFixture
Definition:
ParserFlatbuffersFixture.hpp:36
ParserFlatbuffersFixture.hpp
BOOST_AUTO_TEST_SUITE_END
BOOST_AUTO_TEST_SUITE_END()
ParserFlatbuffersFixture::SetupSingleInputSingleOutput
void SetupSingleInputSingleOutput(const std::string &inputName, const std::string &outputName)
Definition:
ParserFlatbuffersFixture.hpp:93
src
armnnTfLiteParser
test
Softmax.cpp
Generated on Wed Nov 25 2020 09:59:11 for ArmNN by
1.8.13