diff options
author | Francis Murtagh <francis.murtagh@arm.com> | 2021-01-28 14:25:15 +0000 |
---|---|---|
committer | Francis Murtagh <francis.murtagh@arm.com> | 2021-01-28 14:28:51 +0000 |
commit | 7909c53e3b91f96a12f2d587741575d4d1becdce (patch) | |
tree | 4b79f3b6256fd98d0521fb60f70a0a7d4c095cd6 /src/armnn/test/UtilsTests.cpp | |
parent | bbc876c0b7046d6cc3aa9a8d64f80a755027d0cf (diff) | |
download | armnn-7909c53e3b91f96a12f2d587741575d4d1becdce.tar.gz |
IVGCVSW-4874 Provide LayerSupportHandle to frontend users
* Add test for new IsBackendRegistered member function of Handle
* Move deprecated messages to new frontend API of LayerSupportHandle
* Update delegate to use dot operator for IsXXXLayerSupported
Signed-off-by: Francis Murtagh <francis.murtagh@arm.com>
Change-Id: I70d7166e207a10e4b3583a827ca0dda2169bcba1
!android-nn-driver:4940
Diffstat (limited to 'src/armnn/test/UtilsTests.cpp')
-rw-r--r-- | src/armnn/test/UtilsTests.cpp | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/src/armnn/test/UtilsTests.cpp b/src/armnn/test/UtilsTests.cpp index fb078de32f..7776a2d3cf 100644 --- a/src/armnn/test/UtilsTests.cpp +++ b/src/armnn/test/UtilsTests.cpp @@ -5,6 +5,7 @@ #include <boost/test/unit_test.hpp> +#include <armnn/BackendHelper.hpp> #include <armnn/Utils.hpp> #include <armnn/Types.hpp> #include <armnn/TypesUtils.hpp> @@ -266,4 +267,17 @@ BOOST_AUTO_TEST_CASE(PermuteQuantizationDim) BOOST_CHECK(permuted.GetQuantizationDim().value() == 1U); } +#if defined(ARMNNREF_ENABLED) +BOOST_AUTO_TEST_CASE(LayerSupportHandle) +{ + auto layerSupportObject = armnn::GetILayerSupportByBackendId("CpuRef"); + armnn::TensorInfo input; + std::string reasonIfUnsupported; + // InputLayer always supported for CpuRef + BOOST_CHECK_EQUAL(layerSupportObject.IsInputSupported(input, reasonIfUnsupported), true); + + BOOST_CHECK(layerSupportObject.IsBackendRegistered()); +} +#endif + BOOST_AUTO_TEST_SUITE_END() |