Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for visible_device_list (0.18 sec)

  1. tensorflow/compiler/jit/xla_platform_info.cc

    }
    
    absl::StatusOr<std::optional<std::set<int>>> ParseVisibleDeviceList(
        absl::string_view visible_device_list) {
      std::set<int> gpu_ids;
      if (visible_device_list.empty()) {
        return {{std::nullopt}};
      }
      const std::vector<string> visible_devices =
          absl::StrSplit(visible_device_list, ',');
      for (const string& platform_device_id_str : visible_devices) {
        int32_t platform_device_id;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 17:23:27 UTC 2024
    - 17.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/xla_platform_info.h

      void operator=(const XlaPlatformInfo&) = delete;
    };
    
    // Returns a set containing the device ids contained in visible_device_list or
    // nullopt if it is empty. It returns error in case of malformed configuration
    // string.
    absl::StatusOr<std::optional<std::set<int>>> ParseVisibleDeviceList(
        absl::string_view visible_device_list);
    
    // Returns the device type for building a DeviceCompiler from the given platform
    // type.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 09:53:30 UTC 2024
    - 7.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/xla_gpu_device.cc

          iter->second == 0) {
        // Device count for GPU is 0.
        return absl::OkStatus();
      }
    
      string allowed_gpus =
          session_options.config.gpu_options().visible_device_list();
      std::optional<std::set<int>> gpu_ids =
          ParseVisibleDeviceList(allowed_gpus).value();
      if (!gpu_ids) {
        gpu_ids.emplace();
        // Fill the gpu_ids set with all devices if config string is empty.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 6.6K bytes
    - Viewed (0)
Back to top