diff --git a/include/aidge/backend/cpu/operator/GatherImpl_forward_kernels.hpp b/include/aidge/backend/cpu/operator/GatherImpl_forward_kernels.hpp index 7cb1f6c04d503e94f7b8862c2266be5466a8896e..0d312e3c143720c7d920128c8d484d4c68439a24 100644 --- a/include/aidge/backend/cpu/operator/GatherImpl_forward_kernels.hpp +++ b/include/aidge/backend/cpu/operator/GatherImpl_forward_kernels.hpp @@ -40,7 +40,7 @@ void GatherImpl_cpu_forward_kernel(const typename Gather_Op::Attrs& attrs, const preAxisElems *= inputDims[i]; } - const std::vector<std::int32_t> indices = std::get<0>(attrs); + const std::vector<std::int64_t> indices = std::get<0>(attrs); for (std::size_t i=0; i<preAxisElems; ++i) { for(std::size_t j=0; j<indices.size(); ++j) diff --git a/include/aidge/backend/cpu/operator/SliceImpl_forward_kernels.hpp b/include/aidge/backend/cpu/operator/SliceImpl_forward_kernels.hpp index c18a164be9e0cc66f868c4814f72cb45e35d1f69..d92e9008aff2a4e3c9e392fcc51871001020ce5a 100644 --- a/include/aidge/backend/cpu/operator/SliceImpl_forward_kernels.hpp +++ b/include/aidge/backend/cpu/operator/SliceImpl_forward_kernels.hpp @@ -32,9 +32,9 @@ void SliceImpl_cpu_forward_kernel(const typename Slice_Op::Attrs& attrs, DimSize_t nbAxes = std::get<2>(attrs).size(); for (std::size_t i = 0; i < nbAxes; ++i) { // For each slice operation get the params and cast them to size_t - const std::int32_t axis_ = std::get<2>(attrs)[i]; - const std::int32_t start_ = std::get<0>(attrs)[i]; - const std::int32_t end_ = std::get<1>(attrs)[i]; + const std::int64_t axis_ = std::get<2>(attrs)[i]; + const std::int64_t start_ = std::get<0>(attrs)[i]; + const std::int64_t end_ = std::get<1>(attrs)[i]; const std::size_t axis = axis_ >= 0 ? axis_ : static_cast<std::size_t>(axis_) + inputDims.size(); const std::size_t start = start_ >= 0 ? start_ : start_ + inputDims[axis]; const std::size_t end = end_ >= 0 ? end_ : end_ + inputDims[axis];