MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / tensor_to_dlpack

Function tensor_to_dlpack

imperative/python/src/tensor.cpp:806–810  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

804}
805
806PyObject* tensor_to_dlpack(PyObject* tensor) {
807 TensorWrapper* wrapper = TensorWrapper::try_cast(tensor);
808 DLManagedTensor* dlMTensor = to_dlpack(wrapper->m_tensor->data());
809 return PyCapsule_New(dlMTensor, "dltensor", dlpack_capsule_destructor);
810}
811
812PyObject* tensor_from_dlpack(PyObject* data, PyObject* stream) {
813 DLManagedTensor* dlMTensor =

Callers 1

init_tensorFunction · 0.85

Calls 3

try_castFunction · 0.85
to_dlpackFunction · 0.85
dataMethod · 0.45

Tested by

no test coverage detected