1 //
2 // Copyright © 2019 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5
6 #include <boost/test/unit_test.hpp>
7 #include "ParserFlatbuffersFixture.hpp"
8 #include "../TfLiteParser.hpp"
9
10 #include <string>
11 #include <iostream>
12
13 using armnnTfLiteParser::TfLiteParser;
14
15 BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
16
17 struct ResizeNearestNeighborFixture : public ParserFlatbuffersFixture
18 {
ResizeNearestNeighborFixtureResizeNearestNeighborFixture19 explicit ResizeNearestNeighborFixture(const std::string & inputShape,
20 const std::string & outputShape,
21 const std::string & sizeShape,
22 const std::string & sizeData)
23 {
24 m_JsonString = R"(
25 {
26 "version": 3,
27 "operator_codes": [ { "builtin_code": "RESIZE_NEAREST_NEIGHBOR" } ],
28 "subgraphs": [ {
29 "tensors": [
30 {
31 "shape": )" + sizeShape + R"( ,
32 "type": "INT32",
33 "buffer": 0,
34 "name": "sizeTensor",
35 "quantization": {
36 "min": [ 0.0 ],
37 "max": [ 255.0 ],
38 "scale": [ 1.0 ],
39 "zero_point": [ 0 ],
40 }
41 },
42 {
43 "shape": )" + inputShape + R"(,
44 "type": "FLOAT32",
45 "buffer": 1,
46 "name": "InputTensor",
47 "quantization": {
48 "min": [ 0.0 ],
49 "max": [ 255.0 ],
50 "scale": [ 1.0 ],
51 "zero_point": [ 0 ],
52 }
53 },
54 {
55 "shape": )" + outputShape + R"( ,
56 "type": "FLOAT32",
57 "buffer": 2,
58 "name": "OutputTensor",
59 "quantization": {
60 "min": [ 0.0 ],
61 "max": [ 255.0 ],
62 "scale": [ 1.0 ],
63 "zero_point": [ 0 ],
64 }
65 }
66 ],
67 "inputs": [ 1 ],
68 "outputs": [ 2 ],
69 "operators": [
70 {
71 "opcode_index": 0,
72 "inputs": [ 1, 0 ],
73 "outputs": [ 2 ],
74 "builtin_options_type": "ResizeNearestNeighborOptions",
75 "builtin_options": {
76 },
77 "custom_options_format": "FLEXBUFFERS"
78 }
79 ],
80 } ],
81 "buffers" : [
82 { "data": )" + sizeData + R"(, },
83 { },
84 { },
85 ]
86 }
87 )";
88 Setup();
89 }
90 };
91
92
93 struct SimpleResizeNearestNeighborFixture : ResizeNearestNeighborFixture
94 {
SimpleResizeNearestNeighborFixtureSimpleResizeNearestNeighborFixture95 SimpleResizeNearestNeighborFixture()
96 : ResizeNearestNeighborFixture("[ 1, 2, 2, 1 ]", // inputShape
97 "[ 1, 1, 1, 1 ]", // outputShape
98 "[ 2 ]", // sizeShape
99 "[ 1,0,0,0, 1,0,0,0 ]") // sizeData
100 {}
101 };
102
BOOST_FIXTURE_TEST_CASE(ParseResizeNearestNeighbor,SimpleResizeNearestNeighborFixture)103 BOOST_FIXTURE_TEST_CASE(ParseResizeNearestNeighbor, SimpleResizeNearestNeighborFixture)
104 {
105 RunTest<4, armnn::DataType::Float32>(
106 0,
107 {{"InputTensor", { 1.0f, 2.0f, 3.0f, 4.0f }}},
108 {{"OutputTensor", { 1.0f }}});
109 }
110
111 BOOST_AUTO_TEST_SUITE_END()
112