diff options
| author | Matthew Sloyan <matthew.sloyan@arm.com> | 2020-09-14 11:47:26 +0100 |
|---|---|---|
| committer | Jan Eilers <jan.eilers@arm.com> | 2020-09-17 09:41:12 +0000 |
| commit | 0663d66c64d8f1cb2b6158f9018c9ba19e2c504b (patch) | |
| tree | ab8370395d7009f109e15463dbcf644191c1d639 /src/armnn/layers/MeanLayer.cpp | |
| parent | 171214c8ff275c90cd4f7fc23a34ec2c83b5ea39 (diff) | |
| download | armnn-0663d66c64d8f1cb2b6158f9018c9ba19e2c504b.tar.gz | |
IVGCVSW-5303 Remove some boost::numeric_cast from rest of ArmNN
* Replaced with armnn/utility/NumericCast.hpp
* Exclusions in TypeUtils.cpp and QuantizerVisitor.cpp
* Excluded as requires float implementation in NumericCast.hpp
Signed-off-by: Matthew Sloyan <matthew.sloyan@arm.com>
Change-Id: I5c4c60e7028e1a51bf9379457278d253fd37bc70
Diffstat (limited to 'src/armnn/layers/MeanLayer.cpp')
| -rw-r--r-- | src/armnn/layers/MeanLayer.cpp | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/armnn/layers/MeanLayer.cpp b/src/armnn/layers/MeanLayer.cpp index a1a3a40d95..0c5959ca0d 100644 --- a/src/armnn/layers/MeanLayer.cpp +++ b/src/armnn/layers/MeanLayer.cpp @@ -6,6 +6,8 @@ #include "MeanLayer.hpp" #include "LayerCloneBase.hpp" +#include <armnn/utility/NumericCast.hpp> + #include <backendsCommon/CpuTensorHandle.hpp> #include <backendsCommon/WorkloadData.hpp> #include <backendsCommon/WorkloadFactory.hpp> @@ -69,7 +71,7 @@ void MeanLayer::ValidateTensorShapesFromInputs() } else { - outputRank = input.GetNumDimensions() - boost::numeric_cast<unsigned int>(m_Param.m_Axis.size()); + outputRank = input.GetNumDimensions() - armnn::numeric_cast<unsigned int>(m_Param.m_Axis.size()); if (outputRank == 0) { outputRank = 1; @@ -85,7 +87,7 @@ void MeanLayer::ValidateTensorShapesFromInputs() { if (std::find(m_Param.m_Axis.begin(), m_Param.m_Axis.end(), i) == m_Param.m_Axis.end()) { - dimSizes[outputIndex] = boost::numeric_cast<unsigned int>(input.GetShape()[i]); + dimSizes[outputIndex] = armnn::numeric_cast<unsigned int>(input.GetShape()[i]); ++outputIndex; } else if (m_Param.m_KeepDims) |