diff options
Diffstat (limited to 'src/armnn')
-rw-r--r-- | src/armnn/layers/MeanLayer.cpp | 2 | ||||
-rw-r--r-- | src/armnn/test/UtilsTests.cpp | 7 |
2 files changed, 7 insertions, 2 deletions
diff --git a/src/armnn/layers/MeanLayer.cpp b/src/armnn/layers/MeanLayer.cpp index c72d79bab9..c925a3e18f 100644 --- a/src/armnn/layers/MeanLayer.cpp +++ b/src/armnn/layers/MeanLayer.cpp @@ -45,7 +45,7 @@ void MeanLayer::ValidateTensorShapesFromInputs() const TensorInfo& input = GetInputSlot(0).GetConnection()->GetTensorInfo(); - BOOST_ASSERT_MSG(input.GetNumDimensions() > 0 && input.GetNumDimensions() <= MaxNumOfTensorDimensions, + BOOST_ASSERT_MSG(input.GetNumDimensions() > 0 && input.GetNumDimensions() <= 4, "MeanLayer: Mean supports up to 4D input."); unsigned int rank = input.GetNumDimensions(); diff --git a/src/armnn/test/UtilsTests.cpp b/src/armnn/test/UtilsTests.cpp index 0fa67e5f2a..897a35fe4c 100644 --- a/src/armnn/test/UtilsTests.cpp +++ b/src/armnn/test/UtilsTests.cpp @@ -25,7 +25,7 @@ BOOST_AUTO_TEST_CASE(DataTypeSize) BOOST_AUTO_TEST_CASE(PermuteDescriptorWithTooManyMappings) { - BOOST_CHECK_THROW(armnn::PermuteDescriptor({ 0u, 1u, 2u, 3u, 4u }), armnn::InvalidArgumentException); + BOOST_CHECK_THROW(armnn::PermuteDescriptor({ 0u, 1u, 2u, 3u, 4u, 5u }), armnn::InvalidArgumentException); } BOOST_AUTO_TEST_CASE(PermuteDescriptorWithInvalidMappings1d) @@ -48,6 +48,11 @@ BOOST_AUTO_TEST_CASE(PermuteDescriptorWithInvalidMappings4d) BOOST_CHECK_THROW(armnn::PermuteDescriptor({ 0u, 1u, 2u, 4u }), armnn::InvalidArgumentException); } +BOOST_AUTO_TEST_CASE(PermuteDescriptorWithInvalidMappings5d) +{ + BOOST_CHECK_THROW(armnn::PermuteDescriptor({ 0u, 1u, 2u, 3u, 5u }), armnn::InvalidArgumentException); +} + BOOST_AUTO_TEST_CASE(PermuteDescriptorWithDuplicatedMappings) { BOOST_CHECK_THROW(armnn::PermuteDescriptor({ 1u, 1u, 0u }), armnn::InvalidArgumentException); |