diff options
author | Matthew Sloyan <matthew.sloyan@arm.com> | 2020-09-09 09:07:37 +0100 |
---|---|---|
committer | Jan Eilers <jan.eilers@arm.com> | 2020-09-17 08:31:09 +0000 |
commit | 171214c8ff275c90cd4f7fc23a34ec2c83b5ea39 (patch) | |
tree | 23fd3ee288d631c8c94bede71f89f0f1e12da862 /src/backends/neon/workloads/NeonArgMinMaxWorkload.cpp | |
parent | a25886e0966a6b9433cd23595688fadb88a161b2 (diff) | |
download | armnn-171214c8ff275c90cd4f7fc23a34ec2c83b5ea39.tar.gz |
IVGCVSW-5300 Remove some boost::numeric_cast from armnn/backends
* Replaced with armnn/utility/NumericCast.hpp
* Some exclusions in reference backend
* Excluded as requires float implementation in NumericCast.hpp
Signed-off-by: Matthew Sloyan <matthew.sloyan@arm.com>
Change-Id: I9e4e9cd502c865452128fa04415fd6f250baa855
Diffstat (limited to 'src/backends/neon/workloads/NeonArgMinMaxWorkload.cpp')
-rw-r--r-- | src/backends/neon/workloads/NeonArgMinMaxWorkload.cpp | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/backends/neon/workloads/NeonArgMinMaxWorkload.cpp b/src/backends/neon/workloads/NeonArgMinMaxWorkload.cpp index 0fb819db0b..6290ecce17 100644 --- a/src/backends/neon/workloads/NeonArgMinMaxWorkload.cpp +++ b/src/backends/neon/workloads/NeonArgMinMaxWorkload.cpp @@ -10,6 +10,7 @@ #include <backendsCommon/CpuTensorHandle.hpp> +#include <armnn/utility/NumericCast.hpp> #include <armnn/utility/PolymorphicDowncast.hpp> #include <armnnUtils/TensorUtils.hpp> @@ -36,7 +37,7 @@ arm_compute::Status NeonArgMinMaxWorkloadValidate(const TensorInfo& input, auto numDims = input.GetNumDimensions(); auto unsignedAxis = armnnUtils::GetUnsignedAxis(numDims, descriptor.m_Axis); - int aclAxis = boost::numeric_cast<int>(CalcAclAxis(numDims, unsignedAxis)); + int aclAxis = armnn::numeric_cast<int>(CalcAclAxis(numDims, unsignedAxis)); if (descriptor.m_Function == ArgMinMaxFunction::Max) { @@ -60,7 +61,7 @@ NeonArgMinMaxWorkload::NeonArgMinMaxWorkload(const ArgMinMaxQueueDescriptor& des auto numDims = info.m_InputTensorInfos[0].GetNumDimensions(); auto unsignedAxis = armnnUtils::GetUnsignedAxis(numDims, m_Data.m_Parameters.m_Axis); - int aclAxis = boost::numeric_cast<int>(CalcAclAxis(numDims, unsignedAxis)); + int aclAxis = armnn::numeric_cast<int>(CalcAclAxis(numDims, unsignedAxis)); auto layer = std::make_unique<arm_compute::NEArgMinMaxLayer>(); |