diff --git a/include/aidge/backend/cpu/data/TensorImpl.hpp b/include/aidge/backend/cpu/data/TensorImpl.hpp index 967b42333a8748999f99afcb6e52b9a15bf936e3..590c83ed0cbf3b302e6b45c34ffe52e5f85963d1 100644 --- a/include/aidge/backend/cpu/data/TensorImpl.hpp +++ b/include/aidge/backend/cpu/data/TensorImpl.hpp @@ -72,6 +72,16 @@ static Registrar<Tensor> registrarTensorImpl_cpu_Float32( {"cpu", DataType::Float32}, Aidge::TensorImpl_cpu<float>::create); static Registrar<Tensor> registrarTensorImpl_cpu_Int32( {"cpu", DataType::Int32}, Aidge::TensorImpl_cpu<int>::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_UInt8( + {"cpu", DataType::UInt8}, Aidge::TensorImpl_cpu<uint8_t>::create); + + } // namespace } // namespace Aidge