2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
6 #include <boost/test/unit_test.hpp>
7 #include "ParserFlatbuffersSerializeFixture.hpp"
8 #include <armnnDeserializer/IDeserializer.hpp>
12 BOOST_AUTO_TEST_SUITE(Deserializer)
14 struct BatchToSpaceNdFixture : public ParserFlatbuffersSerializeFixture
16 explicit BatchToSpaceNdFixture(const std::string &inputShape,
17 const std::string &blockShape,
18 const std::string &crops,
19 const std::string &dataLayout,
20 const std::string &outputShape,
21 const std::string &dataType)
29 layer_type: "InputLayer",
35 layerName: "InputLayer",
39 connection: {sourceLayerIndex:0, outputSlotIndex:0 },
44 dimensions: )" + inputShape + R"(,
45 dataType: )" + dataType + R"(
53 layer_type: "BatchToSpaceNdLayer",
57 layerName: "BatchToSpaceNdLayer",
58 layerType: "BatchToSpaceNd",
61 connection: {sourceLayerIndex:0, outputSlotIndex:0 },
66 dimensions: )" + outputShape + R"(,
67 dataType: )" + dataType + R"(
72 blockShape: )" + blockShape + R"(,
73 crops: )" + crops + R"(,
74 dataLayout: )" + dataLayout + R"(,
79 layer_type: "OutputLayer",
85 layerName: "OutputLayer",
89 connection: {sourceLayerIndex:1, outputSlotIndex:0 },
94 dimensions: )" + outputShape + R"(,
95 dataType: )" + dataType + R"(
105 SetupSingleInputSingleOutput("InputLayer", "OutputLayer");
109 struct SimpleBatchToSpaceNdFixture : BatchToSpaceNdFixture
111 SimpleBatchToSpaceNdFixture() : BatchToSpaceNdFixture("[ 4, 2, 2, 1 ]",
119 BOOST_FIXTURE_TEST_CASE(SimpleBatchToSpaceNdFloat32, SimpleBatchToSpaceNdFixture)
121 RunTest<4, armnn::DataType::Float32>(0,
123 1.0f, 3.0f, 9.0f, 11.0f,
124 2.0f, 4.0f, 10.0f, 12.0f,
125 5.0f, 7.0f, 13.0f, 15.0f,
126 6.0f, 8.0f, 14.0f, 16.0f
129 1.0f, 2.0f, 3.0f, 4.0f,
130 5.0f, 6.0f, 7.0f, 8.0f,
131 9.0f, 10.0f, 11.0f, 12.0f,
132 13.0f, 14.0f, 15.0f, 16.0f
136 BOOST_AUTO_TEST_SUITE_END()