1 // 2 // Copyright © 2020 Arm Ltd and Contributors. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 6 #include "ParserFlatbuffersFixture.hpp" 7 8 9 TEST_SUITE("TensorflowLiteParser_LeakyRelu") 10 { 11 struct LeakyReluFixture : public ParserFlatbuffersFixture 12 { LeakyReluFixtureLeakyReluFixture13 explicit LeakyReluFixture() 14 { 15 m_JsonString = R"( 16 { 17 "version": 3, 18 "operator_codes": [ { "builtin_code": "LEAKY_RELU" } ], 19 "subgraphs": [ { 20 "tensors": [ 21 { 22 "shape": [ 1, 7 ], 23 "type": "FLOAT32", 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": "FLOAT32", 36 "buffer": 1, 37 "name": "outputTensor", 38 "quantization": { 39 "min": [ 0.0 ], 40 "max": [ 255.0 ], 41 "scale": [ 1.0 ], 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": "LeakyReluOptions", 54 "builtin_options": { 55 "alpha": 0.01 56 }, 57 "custom_options_format": "FLEXBUFFERS" 58 } 59 ], 60 } ], 61 "buffers" : [ {}, {} ] 62 } 63 )"; 64 SetupSingleInputSingleOutput("inputTensor", "outputTensor"); 65 } 66 }; 67 68 TEST_CASE_FIXTURE(LeakyReluFixture, "ParseLeakyRelu") 69 { 70 RunTest<2, armnn::DataType::Float32>(0, 71 {{ "inputTensor", { -0.1f, -0.2f, -0.3f, -0.4f, 0.1f, 0.2f, 0.3f }}}, 72 {{ "outputTensor", { -0.001f, -0.002f, -0.003f, -0.004f, 0.1f, 0.2f, 0.3f }}}); 73 } 74 75 } 76