diff options
| author | Bruno Goncalves <bruno.slackware@gmail.com> | 2018-12-27 14:20:35 -0200 |
|---|---|---|
| committer | Matthew Bentham <matthew.bentham@arm.com> | 2019-01-22 17:10:06 +0000 |
| commit | 9c761a6335f58901bef01b33f1127f9fae3b2bf3 (patch) | |
| tree | 114970cbde18a4facdc655a2bb917fd06390f168 /src/armnnTfLiteParser/test/Multiplication.cpp | |
| parent | 33f8e3b6c71070fd867809ca6934069a950081dc (diff) | |
| download | armnn-9c761a6335f58901bef01b33f1127f9fae3b2bf3.tar.gz | |
Added AddBroadcastReshapeLayer method to TfLiteParser
Change-Id: I6027f6dcdb3ed23505f0a9c780bd3e3d45d3daff
Diffstat (limited to 'src/armnnTfLiteParser/test/Multiplication.cpp')
| -rw-r--r-- | src/armnnTfLiteParser/test/Multiplication.cpp | 36 |
1 files changed, 36 insertions, 0 deletions
diff --git a/src/armnnTfLiteParser/test/Multiplication.cpp b/src/armnnTfLiteParser/test/Multiplication.cpp index f7e2edd546..dabf868559 100644 --- a/src/armnnTfLiteParser/test/Multiplication.cpp +++ b/src/armnnTfLiteParser/test/Multiplication.cpp @@ -108,4 +108,40 @@ BOOST_FIXTURE_TEST_CASE(ParseMultiplication, SimpleMultiplicationFixture) 45.0f, 50.0f, 55.0f } } }); } +struct MultiplicationBroadcastFixture4D1D : public MultiplicationFixture +{ + MultiplicationBroadcastFixture4D1D() : MultiplicationFixture("[ 1, 2, 2, 3 ]", "[ 1 ]", "[ 1, 2, 2, 3 ]") {} +}; + +BOOST_FIXTURE_TEST_CASE(ParseMultiplicationBroadcast4D1D, MultiplicationBroadcastFixture4D1D) +{ + RunTest<4, float>(0, {{ "inputTensor1", { 0.0f, 1.0f, 2.0f, + 3.0f, 4.0f, 5.0f, + 6.0f, 7.0f, 8.0f, + 9.0f, 10.0f, 11.0f } }, + { "inputTensor2", { 5.0f } } }, + {{ "outputTensor", { 0.0f, 5.0f, 10.0f, + 15.0f, 20.0f, 25.0f, + 30.0f, 35.0f, 40.0f, + 45.0f, 50.0f, 55.0f } } }); +} + +struct MultiplicationBroadcastFixture1D4D : public MultiplicationFixture +{ + MultiplicationBroadcastFixture1D4D() : MultiplicationFixture("[ 1 ]", "[ 1, 2, 2, 3 ]", "[ 1, 2, 2, 3 ]") {} +}; + +BOOST_FIXTURE_TEST_CASE(ParseMultiplicationBroadcast1D4D, MultiplicationBroadcastFixture1D4D) +{ + RunTest<4, float>(0, {{ "inputTensor1", { 3.0f } }, + { "inputTensor2", { 0.0f, 1.0f, 2.0f, + 3.0f, 4.0f, 5.0f, + 6.0f, 7.0f, 8.0f, + 9.0f, 10.0f, 11.0f } } }, + {{ "outputTensor", { 0.0f, 3.0f, 6.0f, + 9.0f, 12.0f, 15.0f, + 18.0f, 21.0f, 24.0f, + 27.0f, 30.0f, 33.0f } } }); +} + BOOST_AUTO_TEST_SUITE_END() |