Searched defs:distributed_client (Results 1 – 3 of 3) sorted by relevance
339 std::shared_ptr<DistributedRuntimeClient> distributed_client, int node_id, in BuildDistributedDevices()424 std::shared_ptr<DistributedRuntimeClient> distributed_client, int node_id, in GetGpuClient()
314 -> StatusOr<std::shared_ptr<PyClient>> { in PYBIND11_MODULE()
68 def make_gpu_client(distributed_client=None, node_id=0, platform_name=None, argument