Searched defs:allowed_devices (Results 1 – 6 of 6) sorted by relevance
30 const absl::optional<std::set<int>>& allowed_devices) in LocalClientOptions()64 const absl::optional<std::set<int>>& allowed_devices) { in set_allowed_devices()69 const absl::optional<std::set<int>>& LocalClientOptions::allowed_devices() in allowed_devices() function in xla::LocalClientOptions
28 const std::vector<string>& allowed_devices, in ReplicateNode()179 const std::vector<string>& allowed_devices, in ReplicateNodes()243 const std::vector<string>& allowed_devices = in ReplicatePerReplicaNodesInFunctionGraph() local
60 const absl::optional<std::set<int>>& allowed_devices) { in set_allowed_devices()65 const absl::optional<std::set<int>>& BackendOptions::allowed_devices() const { in allowed_devices() function in xla::BackendOptions
167 const absl::optional<std::set<int>>& allowed_devices) { in GetStreamExecutors()
125 const absl::optional<std::set<int>>& allowed_devices) { in set_allowed_devices()130 const absl::optional<std::set<int>>& ServiceOptions::allowed_devices() const { in allowed_devices() function in xla::ServiceOptions
137 absl::optional<std::set<int>> allowed_devices; member