// Returns the DLDevice* for the given eager tensor handle.
//
// The caller takes ownership of the returned pointer and is responsible for
// deleting it.
TF_CAPI_EXPORT extern void* TFE_GetDLDevice(TFE_TensorHandle* h,
                                            TF_Status* status);

// Converts eager tensor handle to DLPack (DLManagedTensor*), and return the
// void* for further PyCapsule construction.

    if (shape_arr[i] != 1 && stride_arr[i] != expected_stride) {
      valid = false;
    }
    expected_stride *= shape_arr[i];
  }
  return valid;
}
}  // namespace

void* TFE_GetDLDevice(TFE_TensorHandle* h, TF_Status* status) {
  auto dl_device = GetDlContext(h, status);
  return new DLDevice{dl_device.device_type, dl_device.device_id};
}

void TFE_CallDLManagedTensorDeleter(void* dlm_ptr) {