6 #include <boost/test/unit_test.hpp> 64 Setup({ { "input0", inputShape0 },
65 {
"input1", inputShape1 } },
71 struct SubFixture4D4D :
public SubFixture
73 SubFixture4D4D() : SubFixture({ 1, 2, 2, 3 }, { 1, 2, 2, 3 }) {}
78 RunTest<4>({ {
"input0", { 5.0f, 1.0f, 2.0f,
81 29.0f, 10.0f, 11.0f } },
83 {
"input1", { 0.0f, 1.0f, 3.0f,
86 19.0f, 1.0f, 3.0f } } },
88 { {
"output", { 5.0f, 0.0f, -1.0f,
91 10.0f, 9.0f, 8.0f } } });
94 struct SubBroadcastFixture4D1D :
public SubFixture
96 SubBroadcastFixture4D1D() : SubFixture({ 1, 2, 2, 3 }, { 1 }) {}
101 RunTest<4>({ {
"input0", { 0.0f, 1.0f, 2.0f,
104 9.0f, 10.0f, 11.0f } },
106 {
"input1", { 5.0f } } },
108 { {
"output", { -5.0f, -4.0f, -3.0f,
111 4.0f, 5.0f, 6.0f } } });
114 struct SubBroadcastFixture1D4D :
public SubFixture
116 SubBroadcastFixture1D4D() : SubFixture({ 1 }, { 1, 2, 2, 3 }) {}
121 RunTest<4>({ {
"input0", { 3.0f } },
123 {
"input1", { 0.0f, 1.0f, 2.0f,
126 9.0f, 10.0f, 11.0f } } },
128 { {
"output", { 3.0f, 2.0f, 1.0f,
131 -6.0f, -7.0f, -8.0f } } });
BOOST_FIXTURE_TEST_CASE(SimpleSub, SimpleSubFixture)
BOOST_AUTO_TEST_SUITE_END()
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)