From b95fdae53ac9c5b2a00a3f65731fd23cf44ab307 Mon Sep 17 00:00:00 2001 From: hrouis <houssemeddine.rouis92@gmail.com> Date: Wed, 13 Dec 2023 17:00:56 +0100 Subject: [PATCH] change Slice attr to int64 --- .../backend/cpu/operator/SliceImpl_forward_kernels.hpp | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/include/aidge/backend/cpu/operator/SliceImpl_forward_kernels.hpp b/include/aidge/backend/cpu/operator/SliceImpl_forward_kernels.hpp index 80c036cd..c501a55e 100644 --- a/include/aidge/backend/cpu/operator/SliceImpl_forward_kernels.hpp +++ b/include/aidge/backend/cpu/operator/SliceImpl_forward_kernels.hpp @@ -33,9 +33,9 @@ void SliceImpl_cpu_forward_kernel(const typename Slice_Op::Attrs& attrs, for(std::size_t i=0; i<nbAxes;++i) { // For each slice operation get the params and cast them to size_t - int axis_ = std::get<2>(attrs)[i]; - int start_ = std::get<0>(attrs)[i]; - int end_ = std::get<1>(attrs)[i]; + std::int32_t axis_ = std::get<2>(attrs)[i]; + std::int32_t start_ = std::get<0>(attrs)[i]; + std::int32_t end_ = std::get<1>(attrs)[i]; std::size_t axis = axis_>=0?axis_:axis_+inputDims.size(); std::size_t start = start_>=0?start_:start_+inputDims[axis]; std::size_t end = end_>=0?end_:end_+inputDims[axis]; -- GitLab