Skip to content
Snippets Groups Projects
Commit c4e63a10 authored by Cyril Moineau's avatar Cyril Moineau
Browse files

[TensorImpl] add getRaw method.

parent 447dbc81
No related branches found
No related tags found
1 merge request!6Tensor setter getter
Pipeline #31779 passed
......@@ -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 {
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment