Searched refs:allowed_devices (Results 1 – 12 of 12) sorted by relevance
30 const absl::optional<std::set<int>>& allowed_devices) in LocalClientOptions() argument34 allowed_devices_(allowed_devices) {} in LocalClientOptions()64 const absl::optional<std::set<int>>& allowed_devices) { in set_allowed_devices() argument65 allowed_devices_ = allowed_devices; in set_allowed_devices()69 const absl::optional<std::set<int>>& LocalClientOptions::allowed_devices() in allowed_devices() function in xla::LocalClientOptions111 service_options.set_allowed_devices(options.allowed_devices()); in GetOrCreateLocalClient()
51 const absl::optional<std::set<int>>& allowed_devices = absl::nullopt);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);
60 const absl::optional<std::set<int>>& allowed_devices) { in set_allowed_devices() argument61 allowed_devices_ = allowed_devices; in set_allowed_devices()65 const absl::optional<std::set<int>>& BackendOptions::allowed_devices() const { in allowed_devices() function in xla::BackendOptions88 PlatformUtil::GetStreamExecutors(platform, options.allowed_devices())); in CreateBackend()
167 const absl::optional<std::set<int>>& allowed_devices) { in GetStreamExecutors() argument194 if (allowed_devices && allowed_devices->count(i) == 0) { in GetStreamExecutors()
61 const absl::optional<std::set<int>>& allowed_devices);62 const absl::optional<std::set<int>>& allowed_devices() const;
59 const absl::optional<std::set<int>>& allowed_devices = absl::nullopt);
68 const absl::optional<std::set<int>>& allowed_devices);69 const absl::optional<std::set<int>>& allowed_devices() const;
57 .set_allowed_devices(options.allowed_devices()); in NewService()
124 const absl::optional<std::set<int>>& allowed_devices) { in set_allowed_devices() argument125 allowed_devices_ = allowed_devices; in set_allowed_devices()129 const absl::optional<std::set<int>>& ServiceOptions::allowed_devices() const { in allowed_devices() function in xla::ServiceOptions149 backend_options.set_allowed_devices(options.allowed_devices()); in NewService()
132 absl::optional<std::set<int>> allowed_devices; member
156 options.allowed_devices = gpu_ids; in CreateDevices()
213 allowed_devices_(options.allowed_devices) { in XlaDevice()