From c4e63a10414ecd26007d7a916051b5cc3d8bb198 Mon Sep 17 00:00:00 2001
From: cmoineau <cyril.moineau@cea.fr>
Date: Wed, 20 Sep 2023 06:19:09 +0000
Subject: [PATCH] [TensorImpl] add getRaw method.
---
include/aidge/backend/cpu/data/TensorImpl.hpp | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/include/aidge/backend/cpu/data/TensorImpl.hpp b/include/aidge/backend/cpu/data/TensorImpl.hpp
index dfcb8afa..d39aa450 100644
--- a/include/aidge/backend/cpu/data/TensorImpl.hpp
+++ b/include/aidge/backend/cpu/data/TensorImpl.hpp
@@ -47,6 +47,10 @@ class TensorImpl_cpu : public TensorImpl {
return mData.data();
};
+ void* getRaw(std::size_t idx){
+ return rawPtr() + (idx * sizeof(T));
+ };
+
virtual ~TensorImpl_cpu() = default;
void setRawPtr(void *ptr) override final {
--
GitLab