- Sort Score
- Num 10 results
- Language All
Results 1 - 1 of 1 for TFE_HandleToDLPack (0.15 seconds)
-
tensorflow/c/eager/dlpack.cc
DLManagedTensor* dlMTensor = static_cast<DLManagedTensor*>(dlm_ptr); if (dlMTensor->deleter != nullptr) { dlMTensor->deleter(dlMTensor); } } void* TFE_HandleToDLPack(TFE_TensorHandle* h, TF_Status* status) { auto tf_dlm_context = GetDlContext(h, status); if (!status->status.ok()) { return nullptr; } auto* tf_dlm_data = TFE_TensorHandleDevicePointer(h, status);
Created: Tue Apr 07 12:39:13 GMT 2026 - Last Modified: Thu Mar 13 23:41:52 GMT 2025 - 13K bytes - Click Count (0)