Search Options

Display Count
Sort
Preferred Language
Advanced Search

Results 1 - 1 of 1 for device_ids (0.04 seconds)

The search processing time has exceeded the limit. The displayed results may be partial.

  2. github.com/tensorflow/tensorflow

    tensorflow/c/eager/dlpack.cc 

      DeviceNameUtils::ParsedName parsed_name;
  tensorflow::DeviceNameUtils::ParseFullName(device_name, &parsed_name);
  std::string device_type = parsed_name.type;
  int device_id = 0;
  if (parsed_name.has_id) {
    device_id = parsed_name.id;
  }

  ctx.device_id = device_id;
  if (device_type == "CPU") {
    ctx.device_type = DLDeviceType::kDLCPU;
  } else if (device_type == "GPU") {
#if TENSORFLOW_USE_ROCM
    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