6 #include <boost/test/unit_test.hpp> 8 #include "../Deserializer.hpp" 16 explicit PermuteFixture(
const std::string &inputShape,
17 const std::string &dimMappings,
18 const std::string &outputShape,
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: "PermuteLayer", 55 layerName: "PermuteLayer", 59 connection: {sourceLayerIndex:0, outputSlotIndex:0 }, 64 dimensions: )" + outputShape + R"(, 65 dataType: )" + dataType + R"( 70 dimMappings: )" + dimMappings + R"(, 75 layer_type: "OutputLayer", 81 layerName: "OutputLayer", 85 connection: {sourceLayerIndex:1, outputSlotIndex:0 }, 90 dimensions: )" + outputShape + R"(, 91 dataType: )" + dataType + R"( 105 struct SimplePermute2DFixture : PermuteFixture
107 SimplePermute2DFixture() : PermuteFixture(
"[ 2, 3 ]",
110 "QuantisedAsymm8") {}
115 RunTest<2, armnn::DataType::QAsymmU8>(0,
116 { 1, 2, 3, 4, 5, 6 },
117 { 1, 4, 2, 5, 3, 6 });
120 struct SimplePermute4DFixture : PermuteFixture
122 SimplePermute4DFixture() : PermuteFixture(
"[ 1, 2, 3, 4 ]",
125 "QuantisedAsymm8") {}
130 RunTest<4, armnn::DataType::QAsymmU8>(0,
131 { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12,
132 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24 },
133 { 1, 13, 5, 17, 9, 21, 2, 14, 6, 18, 10, 22,
134 3, 15, 7, 19, 11, 23, 4, 16, 8, 20, 12, 24 });
BOOST_FIXTURE_TEST_CASE(SimplePermute2DQuantisedAsymm8, SimplePermute2DFixture)
void SetupSingleInputSingleOutput(const std::string &inputName, const std::string &outputName)
BOOST_AUTO_TEST_SUITE_END()
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)