6 #include <boost/test/unit_test.hpp> 16 unsigned int concatDim)
180 name: "concat_1/axis" 207 input: "concat_1/axis" 228 name: "concat_2/axis" 255 input: "concat_2/axis" 277 Setup({{ "graphInput0", inputShape0 },
278 {
"graphInput1", inputShape1 },
279 {
"graphInput2", inputShape2 },
280 {
"graphInput3", inputShape3}}, {
"concat_2"});
284 struct ConcatOfConcatsFixtureNCHW : ConcatOfConcatsFixture
286 ConcatOfConcatsFixtureNCHW() : ConcatOfConcatsFixture({ 1, 1, 2, 2 }, { 1, 1, 2, 2 }, { 1, 1, 2, 2 },
287 { 1, 1, 2, 2 }, 1 ) {}
290 struct ConcatOfConcatsFixtureNHWC : ConcatOfConcatsFixture
292 ConcatOfConcatsFixtureNHWC() : ConcatOfConcatsFixture({ 1, 1, 2, 2 }, { 1, 1, 2, 2 }, { 1, 1, 2, 2 },
293 { 1, 1, 2, 2 }, 3 ) {}
298 RunTest<4>({{
"graphInput0", {0.0, 1.0, 2.0, 3.0}},
299 {
"graphInput1", {4.0, 5.0, 6.0, 7.0}},
300 {
"graphInput2", {8.0, 9.0, 10.0, 11.0}},
301 {
"graphInput3", {12.0, 13.0, 14.0, 15.0}}},
302 {{
"concat_2", { 0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0,
303 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0 }}});
308 RunTest<4>({{
"graphInput0", {0.0, 1.0, 2.0, 3.0}},
309 {
"graphInput1", {4.0, 5.0, 6.0, 7.0}},
310 {
"graphInput2", {8.0, 9.0, 10.0, 11.0}},
311 {
"graphInput3", {12.0, 13.0, 14.0, 15.0}}},
312 {{
"concat_2", { 0.0, 1.0, 4.0, 5.0, 8.0, 9.0, 12.0, 13.0,
313 2.0, 3.0, 6.0, 7.0, 10.0, 11.0, 14.0, 15.0 }}});
BOOST_FIXTURE_TEST_CASE(ParseConcatOfConcatsNCHW, ConcatOfConcatsFixtureNCHW)
BOOST_AUTO_TEST_SUITE_END()
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)