diff options
author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-10-01 18:35:44 +0100 |
---|---|---|
committer | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-10-02 15:00:53 +0100 |
commit | f97f6da835802187da03f597dcc30386c7b9b15b (patch) | |
tree | c936cb6f4c38e19f39bfc935aadff0219d6a38b8 /src/backends/neon | |
parent | 680f9911d9d9b369fe321ee2dad014012fb5b20f (diff) | |
download | armnn-f97f6da835802187da03f597dcc30386c7b9b15b.tar.gz |
IVGCVSW-3738 Add end-to-end layer test for DepthToSpace
* Added end-to-end layer test implementation for DepthToSpace
* Added test to reference, CL and NEON backends for all supported
data types and data layouts
* Extracted common data permutation code into new utility file and
refactored some existing tests to reduce code duplication
* Fixed EndToEndLayerTestImpl template to work with Float16 data
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: Iaf7a0012c520451052b20c37e36dc05fa8314ff6
Diffstat (limited to 'src/backends/neon')
-rw-r--r-- | src/backends/neon/test/NeonEndToEndTests.cpp | 59 |
1 files changed, 51 insertions, 8 deletions
diff --git a/src/backends/neon/test/NeonEndToEndTests.cpp b/src/backends/neon/test/NeonEndToEndTests.cpp index 3296918117..9d7fc9d4b9 100644 --- a/src/backends/neon/test/NeonEndToEndTests.cpp +++ b/src/backends/neon/test/NeonEndToEndTests.cpp @@ -8,6 +8,7 @@ #include <backendsCommon/test/AbsEndToEndTestImpl.hpp> #include <backendsCommon/test/ArithmeticTestImpl.hpp> #include <backendsCommon/test/ConcatEndToEndTestImpl.hpp> +#include <backendsCommon/test/DepthToSpaceEndToEndTestImpl.hpp> #include <backendsCommon/test/DequantizeEndToEndTestImpl.hpp> #include <backendsCommon/test/PreluEndToEndTestImpl.hpp> #include <backendsCommon/test/QuantizedLstmEndToEndTestImpl.hpp> @@ -143,6 +144,48 @@ BOOST_AUTO_TEST_CASE(NeonConcatEndToEndDim3Uint8Test) ConcatDim3EndToEnd<armnn::DataType::QuantisedAsymm8>(defaultBackends); } +// DepthToSpace +BOOST_AUTO_TEST_CASE(DephtToSpaceEndToEndNchwFloat32) +{ + DepthToSpaceEndToEnd<armnn::DataType::Float32>(defaultBackends, armnn::DataLayout::NCHW); +} + +BOOST_AUTO_TEST_CASE(DephtToSpaceEndToEndNchwFloat16) +{ + DepthToSpaceEndToEnd<armnn::DataType::Float16>(defaultBackends, armnn::DataLayout::NCHW); +} + +BOOST_AUTO_TEST_CASE(DephtToSpaceEndToEndNchwUint8) +{ + DepthToSpaceEndToEnd<armnn::DataType::QuantisedAsymm8>(defaultBackends, armnn::DataLayout::NCHW); +} + +BOOST_AUTO_TEST_CASE(DephtToSpaceEndToEndNchwInt16) +{ + DepthToSpaceEndToEnd<armnn::DataType::QuantisedSymm16>(defaultBackends, armnn::DataLayout::NCHW); +} + +BOOST_AUTO_TEST_CASE(DephtToSpaceEndToEndNhwcFloat32) +{ + DepthToSpaceEndToEnd<armnn::DataType::Float32>(defaultBackends, armnn::DataLayout::NHWC); +} + +BOOST_AUTO_TEST_CASE(DephtToSpaceEndToEndNhwcFloat16) +{ + DepthToSpaceEndToEnd<armnn::DataType::Float16>(defaultBackends, armnn::DataLayout::NHWC); +} + +BOOST_AUTO_TEST_CASE(DephtToSpaceEndToEndNhwcUint8) +{ + DepthToSpaceEndToEnd<armnn::DataType::QuantisedAsymm8>(defaultBackends, armnn::DataLayout::NHWC); +} + +BOOST_AUTO_TEST_CASE(DephtToSpaceEndToEndNhwcInt16) +{ + DepthToSpaceEndToEnd<armnn::DataType::QuantisedSymm16>(defaultBackends, armnn::DataLayout::NHWC); +} + +// Dequantize BOOST_AUTO_TEST_CASE(DequantizeEndToEndSimpleTest) { DequantizeEndToEndSimple<armnn::DataType::QuantisedAsymm8>(defaultBackends); @@ -163,24 +206,24 @@ BOOST_AUTO_TEST_CASE(NeonPreluEndToEndTestUint8Test) PreluEndToEndPositiveTest<armnn::DataType::QuantisedAsymm8>(defaultBackends); } -BOOST_AUTO_TEST_CASE(NeonSpaceToDepthNHWCEndToEndTest1) +BOOST_AUTO_TEST_CASE(NeonSpaceToDepthNhwcEndToEndTest1) { - SpaceToDepthNHWCEndToEndTest1(defaultBackends); + SpaceToDepthNhwcEndToEndTest1(defaultBackends); } -BOOST_AUTO_TEST_CASE(NeonSpaceToDepthNCHWEndToEndTest1) +BOOST_AUTO_TEST_CASE(NeonSpaceToDepthNchwEndToEndTest1) { - SpaceToDepthNCHWEndToEndTest1(defaultBackends); + SpaceToDepthNchwEndToEndTest1(defaultBackends); } -BOOST_AUTO_TEST_CASE(NeonSpaceToDepthNHWCEndToEndTest2) +BOOST_AUTO_TEST_CASE(NeonSpaceToDepthNhwcEndToEndTest2) { - SpaceToDepthNHWCEndToEndTest2(defaultBackends); + SpaceToDepthNhwcEndToEndTest2(defaultBackends); } -BOOST_AUTO_TEST_CASE(NeonSpaceToDepthNCHWEndToEndTest2) +BOOST_AUTO_TEST_CASE(NeonSpaceToDepthNchwEndToEndTest2) { - SpaceToDepthNCHWEndToEndTest2(defaultBackends); + SpaceToDepthNchwEndToEndTest2(defaultBackends); } BOOST_AUTO_TEST_CASE(NeonSplitter1dEndToEndTest) |