diff options
Diffstat (limited to 'src/armnn/layers/MeanLayer.cpp')
-rw-r--r-- | src/armnn/layers/MeanLayer.cpp | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/src/armnn/layers/MeanLayer.cpp b/src/armnn/layers/MeanLayer.cpp index 01f1133c5c..3259d95afd 100644 --- a/src/armnn/layers/MeanLayer.cpp +++ b/src/armnn/layers/MeanLayer.cpp @@ -6,9 +6,9 @@ #include "MeanLayer.hpp" #include "LayerCloneBase.hpp" -#include "backends/CpuTensorHandle.hpp" -#include "backends/WorkloadData.hpp" -#include "backends/WorkloadFactory.hpp" +#include <backends/CpuTensorHandle.hpp> +#include <backends/WorkloadData.hpp> +#include <backends/WorkloadFactory.hpp> #include <cstring> @@ -44,7 +44,7 @@ void MeanLayer::ValidateTensorShapesFromInputs() VerifyLayerConnections(1, CHECK_LOCATION()); const TensorInfo& input = GetInputSlot(0).GetConnection()->GetTensorInfo(); - + BOOST_ASSERT_MSG(input.GetNumDimensions() > 0 && input.GetNumDimensions() <= MaxNumOfTensorDimensions, "MeanLayer: Mean supports up to 4D input."); |