Function GetDLDevice

dali/python/backend_impl.cc:549–554

template <typename Backend>
DLDevice GetDLDevice(const Tensor<Backend> &tensor) {
  if constexpr (std::is_same_v<Backend, GPUBackend>)
    return { kDLCUDA, tensor.device_id() };
  else
    return { tensor.is_pinned() ? kDLCUDAHost : kDLCPU };
}

template <typename Backend>
DLMTensorPtr ToDLMTensor(Tensor<Backend> &tensor,