6 #include <boost/test/unit_test.hpp> 8 #include "../Deserializer.hpp" 16 explicit ReduceSumFixture(
const std::string& inputShape,
17 const std::string& outputShape,
18 const std::string& axis,
19 const std::string& dataType)
27 layer_type: "InputLayer", 33 layerName: "InputLayer", 37 connection: {sourceLayerIndex:0, outputSlotIndex:0 }, 42 dimensions: )" + inputShape + R"(, 43 dataType: )" + dataType + R"( 51 layer_type: "ReduceLayer", 55 layerName: "ReduceSumLayer", 59 connection: {sourceLayerIndex:0, outputSlotIndex:0 }, 64 dimensions: )" + outputShape + R"(, 65 dataType: )" + dataType + R"( 70 axis: )" + axis + R"(, 77 layer_type: "OutputLayer", 83 layerName: "OutputLayer", 87 connection: {sourceLayerIndex:1, outputSlotIndex:0 }, 92 dimensions: )" + outputShape + R"(, 93 dataType: )" + dataType + R"( 107 struct SimpleReduceSumFixture : ReduceSumFixture
109 SimpleReduceSumFixture()
110 : ReduceSumFixture(
"[ 1, 1, 3, 2 ]",
119 RunTest<4, armnn::DataType::Float32>(
121 {{
"InputLayer", { 1.0f, 1.0f, 2.0f, 2.0f, 3.0f, 3.0f }}},
122 {{
"OutputLayer", { 6.0f, 6.0f }}});
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
BOOST_AUTO_TEST_SUITE_END()
BOOST_FIXTURE_TEST_CASE(SimpleReduceSum, SimpleReduceSumFixture)