xref: /aosp_15_r20/external/armnn/src/armnnTfLiteParser/test/Softmax.cpp (revision 89c4ff92f2867872bb9e2354d150bf0c8c502810)
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 {
SoftmaxFixtureSoftmaxFixture13     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 }
74