From 6a2e2fb5bdeec564d64e31244b90d54333d8c5a1 Mon Sep 17 00:00:00 2001 From: ms245755 <michal.szczepanski@cea.fr> Date: Mon, 8 Jul 2024 14:04:43 +0000 Subject: [PATCH] fix tensorImpl --- include/aidge/backend/cpu/data/TensorImpl.hpp | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/include/aidge/backend/cpu/data/TensorImpl.hpp b/include/aidge/backend/cpu/data/TensorImpl.hpp index ff5a4fc4b..6454ed233 100644 --- a/include/aidge/backend/cpu/data/TensorImpl.hpp +++ b/include/aidge/backend/cpu/data/TensorImpl.hpp @@ -132,10 +132,14 @@ static Registrar<Tensor> registrarTensorImpl_cpu_Int32( {"cpu", DataType::Int32}, Aidge::TensorImpl_cpu<int32_t>::create); static Registrar<Tensor> registrarTensorImpl_cpu_Int16( {"cpu", DataType::Int16}, Aidge::TensorImpl_cpu<int16_t>::create); -static Registrar<Tensor> registrarTensorImpl_cpu_UInt16( - {"cpu", DataType::UInt16}, Aidge::TensorImpl_cpu<uint16_t>::create); static Registrar<Tensor> registrarTensorImpl_cpu_Int8( {"cpu", DataType::Int8}, Aidge::TensorImpl_cpu<int8_t>::create); +static Registrar<Tensor> registrarTensorImpl_cpu_UInt64( + {"cpu", DataType::UInt64}, Aidge::TensorImpl_cpu<uint64_t>::create); +static Registrar<Tensor> registrarTensorImpl_cpu_UInt32( + {"cpu", DataType::UInt32}, Aidge::TensorImpl_cpu<uint32_t>::create); +static Registrar<Tensor> registrarTensorImpl_cpu_UInt16( + {"cpu", DataType::UInt16}, Aidge::TensorImpl_cpu<uint16_t>::create); static Registrar<Tensor> registrarTensorImpl_cpu_UInt8( {"cpu", DataType::UInt8}, Aidge::TensorImpl_cpu<uint8_t>::create); } // namespace -- GitLab