Skip to content
Snippets Groups Projects
Commit ee232ad7 authored by Olivier BICHLER's avatar Olivier BICHLER
Browse files

Added missing getRaw

parent 5bde89f6
No related branches found
No related tags found
1 merge request!1Changes following MR
Pipeline #32480 passed
......@@ -58,7 +58,11 @@ class TensorImpl_cuda : public TensorImpl, public TensorImpl_cuda_ {
void *rawPtr() override {
lazyInit(reinterpret_cast<void**>(&mData));
return mData;
};
}
void* getRaw(std::size_t idx) {
return static_cast<void*>(static_cast<T*>(rawPtr()) + idx);
}
const cudnnTensorDescriptor_t& getCudnnTensorDesc() const override {
if (mCudnnTensor == nullptr) {
......
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