Searched defs:distributed_client (Results 1 – 3 of 3) sorted by relevance
246 std::shared_ptr<DistributedRuntimeClient> distributed_client, int node_id, in BuildDistributedDevices()315 std::shared_ptr<DistributedRuntimeClient> distributed_client, int node_id) { in GetGpuClient()
247 int node_id) -> StatusOr<std::shared_ptr<PyClient>> { in PYBIND11_MODULE()
69 def _gpu_backend_factory(distributed_client=None, node_id=0): argument