11 #include <boost/test/unit_test.hpp> 71 Setup({{"input0", inputShape0 },
72 {
"input1", inputShape1 }}, {
"output"});
76 struct Stack3DFixture : StackFixture
78 Stack3DFixture() : StackFixture({ 3, 2, 3 }, { 3, 2, 3 }, 3 ) {}
84 RunTest<4>({ {
"input0", { 1, 2, 3,
92 {
"input1", { 19, 20, 21,
100 { {
"output", { 1, 19,
125 struct Stack3DNegativeAxisFixture : StackFixture
127 Stack3DNegativeAxisFixture() : StackFixture({ 3, 2, 3 }, { 3, 2, 3 }, -1 ) {}
133 RunTest<4>({ {
"input0", { 1, 2, 3,
141 {
"input1", { 19, 20, 21,
149 { {
"output", { 1, 19,
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
BOOST_AUTO_TEST_SUITE_END()
BOOST_FIXTURE_TEST_CASE(Stack3D, Stack3DFixture)