diff options
Diffstat (limited to 'src/backends/cl/workloads/ClArgMinMaxWorkload.cpp')
| -rw-r--r-- | src/backends/cl/workloads/ClArgMinMaxWorkload.cpp | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/backends/cl/workloads/ClArgMinMaxWorkload.cpp b/src/backends/cl/workloads/ClArgMinMaxWorkload.cpp index a79a7b286d..5910080859 100644 --- a/src/backends/cl/workloads/ClArgMinMaxWorkload.cpp +++ b/src/backends/cl/workloads/ClArgMinMaxWorkload.cpp @@ -11,6 +11,7 @@ #include <backendsCommon/CpuTensorHandle.hpp> #include <armnnUtils/TensorUtils.hpp> +#include <armnn/utility/NumericCast.hpp> #include <cl/ClTensorHandle.hpp> #include <cl/ClLayerSupport.hpp> @@ -36,7 +37,7 @@ arm_compute::Status ClArgMinMaxWorkloadValidate(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 @@ ClArgMinMaxWorkload::ClArgMinMaxWorkload(const ArgMinMaxQueueDescriptor& descrip 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)); if (m_Data.m_Parameters.m_Function == ArgMinMaxFunction::Max) { |