1 // 2 // Copyright © 2022 Arm Ltd and Contributors. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 6 #include "ParserFlatbuffersFixture.hpp" 7 8 9 TEST_SUITE("TensorflowLiteParser_LogSoftmax") 10 { 11 struct LogSoftmaxFixture : public ParserFlatbuffersFixture 12 { LogSoftmaxFixtureLogSoftmaxFixture13 explicit LogSoftmaxFixture() 14 { 15 m_JsonString = R"( 16 { 17 "version": 3, 18 "operator_codes": [ { "builtin_code": "LOG_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": "LogSoftmaxOptions", 54 "custom_options_format": "FLEXBUFFERS" 55 } 56 ], 57 } ], 58 "buffers" : [ {}, {} ] 59 } 60 )"; 61 SetupSingleInputSingleOutput("inputTensor", "outputTensor"); 62 } 63 }; 64 65 TEST_CASE_FIXTURE(LogSoftmaxFixture, "ParseLogSoftmaxLite") 66 { 67 RunTest<2, armnn::DataType::QAsymmU8>(0, { 0, 0, 100, 0, 0, 0, 0 }, { 0, 0, 255, 0, 0, 0, 0 }); 68 } 69 70 } 71