6 #include <boost/test/unit_test.hpp> 8 #include "../Deserializer.hpp" 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"( 109 struct SimpleSpaceToBatchNdFixture : SpaceToBatchNdFixture
111 SimpleSpaceToBatchNdFixture() : SpaceToBatchNdFixture(
"[ 2, 1, 2, 4 ]",
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
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
void SetupSingleInputSingleOutput(const std::string &inputName, const std::string &outputName)
BOOST_FIXTURE_TEST_CASE(SimpleSpaceToBatchNdFloat32, SimpleSpaceToBatchNdFixture)
BOOST_AUTO_TEST_SUITE_END()