21.11
Softmax.cpp
Go to the documentation of this file.
1
//
2
// Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
3
// SPDX-License-Identifier: MIT
4
//
5
6
#include "
ParserFlatbuffersFixture.hpp
"
7
8
9
TEST_SUITE
(
"TensorflowLiteParser_Softmax"
)
10
{
11
struct
SoftmaxFixture :
public
ParserFlatbuffersFixture
12
{
13
explicit
SoftmaxFixture()
14
{
15
m_JsonString = R
"(
16
{
17
"version": 3,
18
"operator_codes": [ { "builtin_code": "SOFTMAX" } ],
19
"subgraphs": [ {
20
"tensors": [
21
{
22
"shape": [ 1, 7 ],
23
"type": "UINT8",
24
"buffer": 0,
25
"name": "inputTensor",
26
"quantization": {
27
"min": [ 0.0 ],
28
"max": [ 255.0 ],
29
"scale": [ 1.0 ],
30
"zero_point": [ 0 ],
31
}
32
},
33
{
34
"shape": [ 1, 7 ],
35
"type": "UINT8",
36
"buffer": 1,
37
"name": "outputTensor",
38
"quantization": {
39
"min": [ 0.0 ],
40
"max": [ 255.0 ],
41
"scale": [ 0.00390625 ],
42
"zero_point": [ 0 ],
43
}
44
}
45
],
46
"inputs": [ 0 ],
47
"outputs": [ 1 ],
48
"operators": [
49
{
50
"opcode_index": 0,
51
"inputs": [ 0 ],
52
"outputs": [ 1 ],
53
"builtin_options_type": "SoftmaxOptions",
54
"builtin_options": {
55
"beta": 1.0
56
},
57
"custom_options_format": "FLEXBUFFERS"
58
}
59
],
60
} ],
61
"buffers" : [ {}, {} ]
62
}
63
)";
64
SetupSingleInputSingleOutput
(
"inputTensor"
,
"outputTensor"
);
65
}
66
};
67
68
TEST_CASE_FIXTURE
(SoftmaxFixture,
"ParseSoftmaxLite"
)
69
{
70
RunTest<2, armnn::DataType::QAsymmU8>(0, { 0, 0, 100, 0, 0, 0, 0 }, { 0, 0, 255, 0, 0, 0, 0 });
71
}
72
73
}
TEST_SUITE
TEST_SUITE("TensorflowLiteParser_Softmax")
Definition:
Softmax.cpp:9
ParserFlatbuffersFixture
Definition:
ParserFlatbuffersFixture.hpp:36
TEST_CASE_FIXTURE
TEST_CASE_FIXTURE(ClContextControlFixture, "CopyBetweenNeonAndGpu")
Definition:
MemCopyTests.cpp:45
ParserFlatbuffersFixture.hpp
ParserFlatbuffersFixture::SetupSingleInputSingleOutput
void SetupSingleInputSingleOutput(const std::string &inputName, const std::string &outputName)
Definition:
ParserFlatbuffersFixture.hpp:153
src
armnnTfLiteParser
test
Softmax.cpp
Generated on Wed Nov 17 2021 12:59:33 for ArmNN by
1.8.13