8 #include "../Deserializer.hpp" 10 #include <boost/test/unit_test.hpp> 18 explicit DepthToSpaceFixture(
const std::string& inputShape,
19 const std::string& outputShape,
20 const std::string& blockSize,
21 const std::string& dataLayout,
22 const std::string& dataType)
30 layer_type: "InputLayer", 36 layerName: "InputLayer", 40 connection: {sourceLayerIndex:0, outputSlotIndex:0 }, 45 dimensions: )" + inputShape + R"(, 46 dataType: )" + dataType + R"( 54 layer_type: "DepthToSpaceLayer", 58 layerName: "DepthToSpaceLayer", 59 layerType: "DepthToSpace", 62 connection: {sourceLayerIndex:0, outputSlotIndex:0 }, 67 dimensions: )" + outputShape + R"(, 68 dataType: )" + dataType + R"( 73 blockSize: )" + blockSize + 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 DepthToSpaceFloat32Fixture : DepthToSpaceFixture
111 DepthToSpaceFloat32Fixture() : DepthToSpaceFixture(
"[ 1, 2, 2, 4 ]",
120 RunTest<4, armnn::DataType::Float32>(
125 9.f, 10.f, 11.f, 12.f,
126 13.f, 14.f, 15.f, 16.f
131 9.f, 10.f, 13.f, 14.f,
132 11.f, 12.f, 15.f, 16.f
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
void SetupSingleInputSingleOutput(const std::string &inputName, const std::string &outputName)
BOOST_FIXTURE_TEST_CASE(DepthToSpaceFloat32, DepthToSpaceFloat32Fixture)
BOOST_AUTO_TEST_SUITE_END()