Home	Sort by relevance Sort by last modified time

Full Search Definition Symbol File Path History   | | Help

    Searched refs:allowed_devices (Results 1 - 12 of 12) sorted by null

/external/tensorflow/tensorflow/compiler/xla/client/
client_library.h	51 const absl::optional<std::set<int>>& allowed_devices = absl::nullopt); 66 // Sets the allowed_devices set for selectively constructing stream executors 69 const absl::optional<std::set<int>>& allowed_devices); 70 const absl::optional<std::set<int>>& allowed_devices() const; 90 const absl::optional<std::set<int>>& allowed_devices = absl::nullopt);
client_library.cc	30 const absl::optional<std::set<int>>& allowed_devices) 34 allowed_devices_(allowed_devices) {} 64 const absl::optional<std::set<int>>& allowed_devices) { 65 allowed_devices_ = allowed_devices; 69 const absl::optional<std::set<int>>& LocalClientOptions::allowed_devices() function in class:xla::LocalClientOptions 111 service_options.set_allowed_devices(options.allowed_devices());
/external/tensorflow/tensorflow/compiler/xla/service/
platform_util.h	64 // If populated, only the devices in allowed_devices will have 71 const absl::optional<std::set<int>>& allowed_devices = absl::nullopt);
backend.h	58 // Sets the allowed_devices for selectively constructing stream executors 61 const absl::optional<std::set<int>>& allowed_devices); 62 const absl::optional<std::set<int>>& allowed_devices() const;
backend.cc	60 const absl::optional<std::set<int>>& allowed_devices) { 61 allowed_devices_ = allowed_devices; 65 const absl::optional<std::set<int>>& BackendOptions::allowed_devices() const { function in class:xla::BackendOptions 110 PlatformUtil::GetStreamExecutors(platform, options.allowed_devices()));
platform_util.cc	213 const absl::optional<std::set<int>>& allowed_devices) { 237 // allowed_devices, we don't make any allocations on other devices. 240 if (allowed_devices && allowed_devices->count(i) == 0) {
service.h	65 // Sets the allowed_devices set for selectively constructing stream executors 68 const absl::optional<std::set<int>>& allowed_devices); 69 const absl::optional<std::set<int>>& allowed_devices() const;
local_service.cc	57 .set_allowed_devices(options.allowed_devices());
service.cc	124 const absl::optional<std::set<int>>& allowed_devices) { 125 allowed_devices_ = allowed_devices; 129 const absl::optional<std::set<int>>& ServiceOptions::allowed_devices() const { function in class:xla::ServiceOptions 149 backend_options.set_allowed_devices(options.allowed_devices());     [all...]
/external/tensorflow/tensorflow/compiler/jit/
xla_gpu_device.cc	101 options.allowed_devices = gpu_ids;
xla_device.h	132 absl::optional<std::set<int>> allowed_devices; member in struct:tensorflow::XlaDevice::Options
xla_device.cc	206 allowed_devices_(options.allowed_devices) {

Completed in 251 milliseconds