diff options
| author | Sadik Armagan <sadik.armagan@arm.com> | 2021-03-25 07:46:55 +0000 |
|---|---|---|
| committer | Sadik Armagan <sadik.armagan@arm.com> | 2021-03-25 07:46:55 +0000 |
| commit | f0a6dec75832604d5ab18242dc216852821a8279 (patch) | |
| tree | ff25e64c62c63975a54abd16a8bff744be70d7c0 /src/armnn/test | |
| parent | 16fb1a2d9c1d3d80c0f0b6ab549919fbabd2a0b9 (diff) | |
| download | armnn-f0a6dec75832604d5ab18242dc216852821a8279.tar.gz | |
IVGCVSW-5736 and IVGCVSW-5743 'NonConstWeights: Update front-end and TfLiteDelegate support for FullyConnected Operator'
* Added front-end support for non-const weights for FULLY_CONNECTED operator
* Added FULLY_CONNECTED end-to-end test
* Updated FULLY_CONNECTED operator support in TfLite Arm NN Delegate for non-const weights
* Updated the version numbers
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: Iffa5b9aa9297aca4c02d923cce4636c88ac21faa
Diffstat (limited to 'src/armnn/test')
| -rw-r--r-- | src/armnn/test/OptimizerTests.cpp | 8 | ||||
| -rw-r--r-- | src/armnn/test/UtilsTests.cpp | 12 |
2 files changed, 20 insertions, 0 deletions
diff --git a/src/armnn/test/OptimizerTests.cpp b/src/armnn/test/OptimizerTests.cpp index fa860abb64..896fdfd68c 100644 --- a/src/armnn/test/OptimizerTests.cpp +++ b/src/armnn/test/OptimizerTests.cpp @@ -10,6 +10,7 @@ #include <Network.hpp> #include <Optimizer.hpp> +#include <armnn/BackendHelper.hpp> #include <armnn/BackendRegistry.hpp> #include <armnn/INetwork.hpp> #include <armnn/LayerVisitorBase.hpp> @@ -679,6 +680,13 @@ public: }; }; +BOOST_AUTO_TEST_CASE(BackendCapabilityTest) +{ + BackendId backendId ="MockBackend"; + // MockBackend does not support the NonConstWeights capability + BOOST_CHECK(!armnn::IsCapabilitySupported(backendId, armnn::BackendCapability::NonConstWeights)); +} + BOOST_AUTO_TEST_CASE(BackendHintTest) { class TestBackendAssignment : public LayerVisitorBase<VisitorNoThrowPolicy> diff --git a/src/armnn/test/UtilsTests.cpp b/src/armnn/test/UtilsTests.cpp index 7776a2d3cf..f0198cb9d4 100644 --- a/src/armnn/test/UtilsTests.cpp +++ b/src/armnn/test/UtilsTests.cpp @@ -278,6 +278,18 @@ BOOST_AUTO_TEST_CASE(LayerSupportHandle) BOOST_CHECK(layerSupportObject.IsBackendRegistered()); } + +BOOST_AUTO_TEST_CASE(IsCapabilitySupported_CpuRef) +{ + BOOST_CHECK(armnn::IsCapabilitySupported(armnn::Compute::CpuRef, armnn::BackendCapability::NonConstWeights)); +} +#endif + +#if defined(ARMCOMPUTENEON_ENABLED) +BOOST_AUTO_TEST_CASE(IsCapabilitySupported_CpuAcc) +{ + BOOST_CHECK(!armnn::IsCapabilitySupported(armnn::Compute::CpuAcc, armnn::BackendCapability::NonConstWeights)); +} #endif BOOST_AUTO_TEST_SUITE_END() |