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