Search Options

Display Count
Sort
Preferred Language
Advanced Search

Results 1 - 2 of 2 for TFE_HandleToDLPack (0.16 seconds)

  1. tensorflow/c/eager/dlpack.h

                                                TF_Status* status);
    
    // Converts eager tensor handle to DLPack (DLManagedTensor*), and return the
    // void* for further PyCapsule construction.
    TF_CAPI_EXPORT extern void* TFE_HandleToDLPack(TFE_TensorHandle* h,
                                                   TF_Status* status);
    
    // Converts DLPack (DLManagedTensor*) to eager tensor handle.
    TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_HandleFromDLPack(void* dlm,
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Thu Mar 13 23:41:52 GMT 2025
    - 1.9K bytes
    - Click Count (0)
  2. 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)
Back to Top