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::BackendOptions110 PlatformUtil::GetStreamExecutors(platform, options.allowed_devices())); in CreateBackend()
213 const absl::optional<std::set<int>>& allowed_devices) { in GetStreamExecutors() argument240 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;
68 const absl::optional<std::set<int>>& allowed_devices);69 const absl::optional<std::set<int>>& allowed_devices() const;
71 const absl::optional<std::set<int>>& allowed_devices = absl::nullopt);
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()
101 options.allowed_devices = gpu_ids; in CreateDevices()
132 absl::optional<std::set<int>> allowed_devices; member
206 allowed_devices_(options.allowed_devices) { in XlaDevice()