diff --git a/include/aidge/backend/cpu/operator/SliceImpl_kernels.hpp b/include/aidge/backend/cpu/operator/SliceImpl_kernels.hpp index 1bf4c491723c570fa8bfd1774beca1630d2de9be..d290c40f26270a789c2d328f98560c65ecac1559 100644 --- a/include/aidge/backend/cpu/operator/SliceImpl_kernels.hpp +++ b/include/aidge/backend/cpu/operator/SliceImpl_kernels.hpp @@ -89,13 +89,13 @@ void SliceImpl_cpu_forward_kernel(const std::vector<std::int64_t>& starts, } REGISTRAR(SliceImpl_cpu, - {DataType::Float32}, + {{DataType::Float32, DataType::Any}, {DataType::Float32}}, {ProdConso::inPlaceModel, Aidge::SliceImpl_cpu_forward_kernel<float, float>, nullptr}); REGISTRAR(SliceImpl_cpu, - {DataType::Float64}, + {{DataType::Float64, DataType::Any}, {DataType::Float64}}, {ProdConso::inPlaceModel, Aidge::SliceImpl_cpu_forward_kernel<double, double>, nullptr}); REGISTRAR(SliceImpl_cpu, - {DataType::Int32}, + {{DataType::Int32, DataType::Any}, {DataType::Int32}}, {ProdConso::inPlaceModel, Aidge::SliceImpl_cpu_forward_kernel<int32_t, int32_t>, nullptr}); } // namespace Aidge