Searched refs:dlm_ptr (Results 1 – 3 of 3) sorted by relevance
37 TF_CAPI_EXPORT extern void TFE_CallDLManagedTensorDeleter(void* dlm_ptr);
244 void TFE_CallDLManagedTensorDeleter(void* dlm_ptr) { in TFE_CallDLManagedTensorDeleter() argument245 DLManagedTensor* dlMTensor = static_cast<DLManagedTensor*>(dlm_ptr); in TFE_CallDLManagedTensorDeleter()
1383 void* dlm_ptr = tensorflow::TFE_HandleToDLPack(thandle, status.get()); in PYBIND11_MODULE() local1387 dlm_ptr, tensorflow::kDlTensorCapsuleName, [](PyObject* capsule) { in PYBIND11_MODULE()