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