MCPcopy Create free account
hub / github.com/NVIDIA/DALI / TensorToDLPackView

Function TensorToDLPackView

dali/util/pybind.h:220–223  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

218
219template <typename Backend>
220py::capsule TensorToDLPackView(SampleView<Backend> tensor, bool pinned, int device_id) {
221 DLMTensorPtr dl_tensor = GetDLTensorView(tensor, pinned, device_id);
222 return DLTensorToCapsule(std::move(dl_tensor));
223}
224
225template <typename Backend>
226py::list TensorListToDLPackView(TensorList<Backend> &tensors) {

Calls 2

GetDLTensorViewFunction · 0.85
DLTensorToCapsuleFunction · 0.85

Tested by

no test coverage detected