-
Notifications
You must be signed in to change notification settings - Fork 609
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Adds data_ptr
method to Tensor and TensorList
#1773
Conversation
data_ptr
method to Tensor and TensorListdata_ptr
method to Tensor and TensorList
CI MESSAGE: [1158767]: BUILD STARTED |
CI MESSAGE: [1158767]: BUILD FAILED |
CI MESSAGE: [1158801]: BUILD STARTED |
CI MESSAGE: [1158801]: BUILD PASSED |
dali/python/backend_impl.cc
Outdated
}, | ||
R"code( | ||
Returns a tensor at given position in the list. | ||
|
||
Parameters | ||
---------- | ||
)code") | ||
)code")l |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
What happened here?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Typo. Fixed.
- adds an ability to expose a raw data pointer form Tensor and TensorList on CPU and GPU - adds tests for `data_ptr` which created numpy or cupy arrays form an exposed pointers Signed-off-by: Janusz Lisiecki <jlisiecki@nvidia.com>
Signed-off-by: Janusz Lisiecki jlisiecki@nvidia.com
Why we need this PR?
Pick one, remove the rest
What happened in this PR?
Fill relevant points, put NA otherwise. Replace anything inside []
adds an ability to expose a raw data pointer form Tensor and TensorList
backend_impl
NA
Tests added
Docstrings are added
In response to #1765
JIRA TASK: [NA]