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