Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for TFE_HandleToDLPack (0.1 sec)

  1. 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);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 12.9K bytes
    - Viewed (0)
Back to top