6 #include <boost/test/unit_test.hpp> 16 explicit RankFixture(
const std::string &inputShape,
17 const std::string &dataType)
25 layer_type: "InputLayer", 37 dimensions: )" + inputShape + R"(, 38 dataType: )" + dataType + R"(, 39 quantizationScale: 0.0 48 layer_type: "RankLayer", 67 quantizationScale: 0.0, 76 layer_type: "OutputLayer", 103 struct SimpleRankDimSize1Fixture : RankFixture
105 SimpleRankDimSize1Fixture() : RankFixture(
"[ 8 ]",
"QSymmS16") {}
108 struct SimpleRankDimSize2Fixture : RankFixture
110 SimpleRankDimSize2Fixture() : RankFixture(
"[ 3, 3 ]",
"QSymmS8") {}
113 struct SimpleRankDimSize3Fixture : RankFixture
115 SimpleRankDimSize3Fixture() : RankFixture(
"[ 2, 2, 1 ]",
"Signed32") {}
118 struct SimpleRankDimSize4Fixture : RankFixture
120 SimpleRankDimSize4Fixture() : RankFixture(
"[ 2, 2, 1, 1 ]",
"Float32") {}
125 RunTest<1, armnn::DataType::QSymmS16, armnn::DataType::Signed32>( 0,
126 { 1, 2, 3, 4, 5, 6, 7, 8 },
132 RunTest<1, armnn::DataType::QSymmS8, armnn::DataType::Signed32>( 0,
133 { 1, 2, 3, 4, 5, 6, 7, 8, 9 },
139 RunTest<1, armnn::DataType::Signed32, armnn::DataType::Signed32>( 0,
146 RunTest<1, armnn::DataType::Float32, armnn::DataType::Signed32>( 0,
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
BOOST_FIXTURE_TEST_CASE(RankDimSize1Float16, SimpleRankDimSize1Fixture)
BOOST_AUTO_TEST_SUITE_END()