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 "../Deserializer.hpp"
12 BOOST_AUTO_TEST_SUITE(Deserializer)
14 struct SpaceToBatchNdFixture : public ParserFlatbuffersSerializeFixture
16 explicit SpaceToBatchNdFixture(const std::string &inputShape,
17 const std::string &blockShape,
18 const std::string &padList,
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: "SpaceToBatchNdLayer",
57 layerName: "SpaceToBatchNdLayer",
58 layerType: "SpaceToBatchNd",
61 connection: {sourceLayerIndex:0, outputSlotIndex:0 },
66 dimensions: )" + outputShape + R"(,
67 dataType: )" + dataType + R"(
72 blockShape: )" + blockShape + R"(,
73 padList: )" + padList + 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 SimpleSpaceToBatchNdFixture : SpaceToBatchNdFixture
111 SimpleSpaceToBatchNdFixture() : SpaceToBatchNdFixture("[ 2, 1, 2, 4 ]",
119 BOOST_FIXTURE_TEST_CASE(SimpleSpaceToBatchNdFloat32, SimpleSpaceToBatchNdFixture)
121 RunTest<4, armnn::DataType::Float32>(0,
123 1.0f, 2.0f, 3.0f, 4.0f,
124 5.0f, 6.0f, 7.0f, 8.0f,
125 9.0f, 10.0f, 11.0f, 12.0f,
126 13.0f, 14.0f, 15.0f, 16.0f
140 BOOST_AUTO_TEST_SUITE_END()