Searched defs:allowed_devices (Results 1 – 10 of 10) sorted by relevance
42 const std::vector<string>& allowed_devices, in ReplicateNode()80 const Edge* edge, const std::vector<string>& allowed_devices, in ReplicateFromCompositeDeviceToCompositeDevice()112 const Edge* edge, const std::vector<string>& allowed_devices, in ReplicateFromCompositeDeviceToRegularDevice()194 Status ReplicateNodesAndEdges(const std::vector<string>& allowed_devices, in ReplicateNodesAndEdges()272 const std::vector<string>& allowed_devices = in ReplicatePerReplicaNodesInFunctionGraph() local
32 const std::optional<std::set<int>>& allowed_devices) in LocalClientOptions()66 const std::optional<std::set<int>>& allowed_devices) { in set_allowed_devices()71 const std::optional<std::set<int>>& LocalClientOptions::allowed_devices() in allowed_devices() function in xla::LocalClientOptions
59 const std::optional<std::set<int>>& allowed_devices) { in set_allowed_devices()64 const std::optional<std::set<int>>& BackendOptions::allowed_devices() const { in allowed_devices() function in xla::BackendOptions
159 const std::optional<std::set<int>>& allowed_devices) { in GetStreamExecutors()
129 const std::optional<std::set<int>>& allowed_devices) { in set_allowed_devices()134 const std::optional<std::set<int>>& ServiceOptions::allowed_devices() const { in allowed_devices() function in xla::ServiceOptions
155 const std::optional<std::set<int>>& allowed_devices) { in GetGpuXlaClient()425 const std::optional<std::set<int>>& allowed_devices, in GetGpuClient()
38 absl::Span<string const> allowed_devices, const char* name, in VarHandleOp()
147 std::optional<std::set<int>> allowed_devices; member
314 -> StatusOr<std::shared_ptr<PyClient>> { in PYBIND11_MODULE()
69 allowed_devices=None): argument