6 #include <boost/test/unit_test.hpp> 14 SplitFixture(
bool withDimZero=
false) {
51 name: "multiplication" 138 Setup( { { "graphInput", { 1, 2, 2 , 2} } , {
"graphInput2", { 1, 2, 2 , 2} }},
139 {
"Relu_1",
"Relu_2" });
143 struct InputFirstSplitFixture : SplitFixture
145 InputFirstSplitFixture() : SplitFixture(
true) {}
151 (m_Parser->GetNetworkOutputBindingInfo(
"Relu_1").second.GetShape() ==
armnn::TensorShape({ 1, 1, 2, 2 })));
154 (m_Parser->GetNetworkOutputBindingInfo(
"Relu_2").second.GetShape() ==
armnn::TensorShape({ 1, 1, 2, 2 })));
156 RunTest<4>({ {
"graphInput", { -1.0f, -0.5f, 1.25f, -3.0f, 0.0f, 0.5f, -0.75f, 1.75f } } },
157 { {
"Relu_1", { 0.0f, 0.0f, 1.25f, 0.0f } },
158 {
"Relu_2", { 0.0f, 0.5f, 0.0f, 1.75f } } });
165 (m_Parser->GetNetworkOutputBindingInfo(
"Relu_1").second.GetShape() ==
armnn::TensorShape({ 1, 2, 2, 1 })));
168 (m_Parser->GetNetworkOutputBindingInfo(
"Relu_2").second.GetShape() ==
armnn::TensorShape({ 1, 2, 2, 1 })));
170 RunTest<4>({ {
"graphInput", { -1.0f, -0.5f, 1.25f, -3.0f, 0.0f, 0.5f, -0.75f , 1.75f } } ,
171 {
"graphInput2", { -1.0f, -0.5f, 1.25f, -3.0f, 0.0f, 0.5f, -0.75f , 1.75f } } },
172 { {
"Relu_1", { 1.0f, 1.5625f, 0, 0.5625f } },
173 {
"Relu_2", { 0.25, 9.0f, 0.25f, 3.0625f } } });
178 SplitLastDimFixture(
bool withDimZero=
false) {
179 boost::ignore_unused(withDimZero);
232 name: "split/split_dim" 255 input: "split/split_dim" 373 Setup( { { "Placeholder", { 1, 2, 2 , 3} } },
374 {
"sub0",
"sub1",
"sub2" });
381 (m_Parser->GetNetworkOutputBindingInfo(
"sub0").second.GetShape() ==
armnn::TensorShape({ 1, 2, 2, 1 })));
384 (m_Parser->GetNetworkOutputBindingInfo(
"sub1").second.GetShape() ==
armnn::TensorShape({ 1, 2, 2, 1 })));
387 (m_Parser->GetNetworkOutputBindingInfo(
"sub2").second.GetShape() ==
armnn::TensorShape({ 1, 2, 2, 1 })));
389 RunTest<4>({ {
"Placeholder", { 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f, 9.0f, 10.0f, 11.0f, 12.0f } } },
390 { {
"sub0", { -2.0f, 1.0f, 4.0f, 7.0f } },
391 {
"sub1", { 0.0f, 3.0f, 6.0f, 9.0f } },
392 {
"sub2", { 2.0f, 5.0f, 8.0f, 11.0f } } });
BOOST_FIXTURE_TEST_CASE(ParseAxisOneSplitTwoFloat32, SimpleSplitFixtureFloat32)
DataLayout::NCHW DataLayout::NCHW DataLayout::NHWC DataLayout::NHWC true
BOOST_AUTO_TEST_SUITE_END()
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)