diff options
Diffstat (limited to 'src/backends/cl/workloads/ClLstmFloatWorkload.cpp')
| -rw-r--r-- | src/backends/cl/workloads/ClLstmFloatWorkload.cpp | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/backends/cl/workloads/ClLstmFloatWorkload.cpp b/src/backends/cl/workloads/ClLstmFloatWorkload.cpp index 9d06428902..fe9b45e054 100644 --- a/src/backends/cl/workloads/ClLstmFloatWorkload.cpp +++ b/src/backends/cl/workloads/ClLstmFloatWorkload.cpp @@ -9,6 +9,8 @@ #include <cl/ClLayerSupport.hpp> #include <aclCommon/ArmComputeTensorUtils.hpp> +#include <armnn/utility/NumericCast.hpp> + #include <arm_compute/runtime/CL/functions/CLLSTMLayer.h> #include "ClWorkloadUtils.hpp" @@ -132,8 +134,8 @@ ClLstmFloatWorkload::ClLstmFloatWorkload(const LstmQueueDescriptor &descriptor, // Get the batch_size and the num_units from the cellStateIn dimensions const TensorInfo& inputTensorInfo = info.m_InputTensorInfos[2]; - const unsigned int batch_size = boost::numeric_cast<unsigned int>(inputTensorInfo.GetShape()[0]); - const unsigned int num_units = boost::numeric_cast<unsigned int>(inputTensorInfo.GetShape()[1]); + const unsigned int batch_size = armnn::numeric_cast<unsigned int>(inputTensorInfo.GetShape()[0]); + const unsigned int num_units = armnn::numeric_cast<unsigned int>(inputTensorInfo.GetShape()[1]); m_ScratchBuffer = std::make_unique<arm_compute::CLTensor>(); if (m_Data.m_Parameters.m_CifgEnabled) |