diff options
| author | David Beck <david.beck@arm.com> | 2018-09-26 17:41:13 +0100 |
|---|---|---|
| committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-10 16:16:57 +0100 |
| commit | ac42efd972b7d03da17f057b2ceaaac5d6e96b1a (patch) | |
| tree | 1ebc1320fa3ea7f494d3716ea79a2bda0f4ffd1e /src/armnn/layers/MeanLayer.cpp | |
| parent | bcd3c85b5a7657b38f503676b88a80ae74165acd (diff) | |
| download | armnn-ac42efd972b7d03da17f057b2ceaaac5d6e96b1a.tar.gz | |
IVGCVSW-1900 : CL backend folder structure
* moving backends/ClWorkloads to backends/cl
* and moving pure Cl workload related code to
backends/cl/workloads
Change-Id: I019a3c6b4da5e7a23074bf03fb057e63199ad129
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."); |