diff options
| author | narpra01 <narumol.prangnawarat@arm.com> | 2018-11-20 15:21:28 +0000 |
|---|---|---|
| committer | Aron Virginas-Tar <aron.virginas-tar@arm.com> | 2018-11-20 15:52:24 +0000 |
| commit | b9546cf1ffde83f63436c4087711dcf098ea4196 (patch) | |
| tree | 9e8bfe040effdc56e39f9cc9d920fb515352f52d /src/backends/reference/test/RefEndToEndTests.cpp | |
| parent | 97f71306bdfdadb7ff7a55043ebce40f6c280223 (diff) | |
| download | armnn-b9546cf1ffde83f63436c4087711dcf098ea4196.tar.gz | |
IVGCVSW-2173 - Add end to end layer test implementation and example usage
* Add CommonTestUtils
* Add end to end layer test implementation
* Add example usage for Merger layer on Ref, Cl, Neon
Change-Id: I8931136288cd68b80bcdad8f5ae087ae1a70a60a
Diffstat (limited to 'src/backends/reference/test/RefEndToEndTests.cpp')
| -rw-r--r-- | src/backends/reference/test/RefEndToEndTests.cpp | 58 |
1 files changed, 48 insertions, 10 deletions
diff --git a/src/backends/reference/test/RefEndToEndTests.cpp b/src/backends/reference/test/RefEndToEndTests.cpp index dc2ffb81ff..97bec51645 100644 --- a/src/backends/reference/test/RefEndToEndTests.cpp +++ b/src/backends/reference/test/RefEndToEndTests.cpp @@ -4,21 +4,22 @@ // #include <backendsCommon/test/EndToEndTestImpl.hpp> +#include <backendsCommon/test/MergerTestImpl.hpp> #include <boost/test/unit_test.hpp> BOOST_AUTO_TEST_SUITE(RefEndToEnd) +std::vector<armnn::BackendId> defaultBackends = {armnn::Compute::CpuRef}; + BOOST_AUTO_TEST_CASE(ConstantUsage_Ref_Float32) { - std::vector<armnn::BackendId> backends = {armnn::Compute::CpuRef}; - BOOST_TEST(ConstantUsageFloat32Test(backends)); + BOOST_TEST(ConstantUsageFloat32Test(defaultBackends)); } BOOST_AUTO_TEST_CASE(ConstantUsage_Ref_Uint8) { - std::vector<armnn::BackendId> backends = {armnn::Compute::CpuRef}; - BOOST_TEST(ConstantUsageUint8Test(backends)); + BOOST_TEST(ConstantUsageUint8Test(defaultBackends)); } BOOST_AUTO_TEST_CASE(Unsigned8) @@ -51,8 +52,7 @@ BOOST_AUTO_TEST_CASE(Unsigned8) softmax->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); // optimize the network - std::vector<armnn::BackendId> backends = {armnn::Compute::CpuRef}; - IOptimizedNetworkPtr optNet = Optimize(*net, backends, runtime->GetDeviceSpec()); + IOptimizedNetworkPtr optNet = Optimize(*net, defaultBackends, runtime->GetDeviceSpec()); // Loads it into the runtime. NetworkId netId; @@ -115,8 +115,7 @@ BOOST_AUTO_TEST_CASE(TrivialAdd) add->GetOutputSlot(0).SetTensorInfo(tensorInfo); // optimize the network - std::vector<armnn::BackendId> backends = {armnn::Compute::CpuRef}; - IOptimizedNetworkPtr optNet = Optimize(*net, backends, runtime->GetDeviceSpec()); + IOptimizedNetworkPtr optNet = Optimize(*net, defaultBackends, runtime->GetDeviceSpec()); // Loads it into the runtime. NetworkId netId; @@ -214,8 +213,7 @@ BOOST_AUTO_TEST_CASE(MultipleOutputs) activation3->GetOutputSlot(0).SetTensorInfo(tensorInfo); // optimize the network - std::vector<armnn::BackendId> backends = {armnn::Compute::CpuRef}; - IOptimizedNetworkPtr optNet = Optimize(*net, backends, runtime->GetDeviceSpec()); + IOptimizedNetworkPtr optNet = Optimize(*net, defaultBackends, runtime->GetDeviceSpec()); // Loads it into the runtime. NetworkId netId; @@ -248,4 +246,44 @@ BOOST_AUTO_TEST_CASE(MultipleOutputs) BOOST_TEST(output3Data == std::vector<float>({ 3.f, 5.f, 2.f, 3.f, 5.f, 2.f, 2.f, 2.f, 3.f, 3.f })); // [2, 5] } +BOOST_AUTO_TEST_CASE(RefMergerEndToEndDim0Test) +{ + MergerDim0EndToEnd<float>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefMergerEndToEndDim0Uint8Test) +{ + MergerDim0EndToEnd<uint8_t>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefMergerEndToEndDim1Test) +{ + MergerDim1EndToEnd<float>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefMergerEndToEndDim1Uint8Test) +{ + MergerDim1EndToEnd<uint8_t>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefMergerEndToEndDim2Test) +{ + MergerDim2EndToEnd<float>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefMergerEndToEndDim2Uint8Test) +{ + MergerDim2EndToEnd<uint8_t>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefMergerEndToEndDim3Test) +{ + MergerDim3EndToEnd<float>(defaultBackends); +} + +BOOST_AUTO_TEST_CASE(RefMergerEndToEndDim3Uint8Test) +{ + MergerDim3EndToEnd<uint8_t>(defaultBackends); +} + BOOST_AUTO_TEST_SUITE_END()
\ No newline at end of file |