- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for visible_device_list (0.18 sec)
-
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) -
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) -
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)