Searched refs:distributed_client (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | gpu_device.cc | 246 std::shared_ptr<DistributedRuntimeClient> distributed_client, int node_id, in BuildDistributedDevices() argument
266 distributed_client->EnumerateDevices(local_topology, &global_topology)); in BuildDistributedDevices()
297 node_id, distributed_client, device_to_node); in BuildDistributedDevices()
315 std::shared_ptr<DistributedRuntimeClient> distributed_client, int node_id) { in GetGpuClient() argument
328 if (distributed_client) { in GetGpuClient()
330 std::move(local_device_states), std::move(distributed_client), node_id, in GetGpuClient()
|
| D | gpu_device.h | 59 std::shared_ptr<DistributedRuntimeClient> distributed_client, int node_id);
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 69 def _gpu_backend_factory(distributed_client=None, node_id=0): argument
92 distributed_client=distributed_client,
|
| D | xla.cc | 246 std::shared_ptr<DistributedRuntimeClient> distributed_client, in PYBIND11_MODULE()
251 std::move(distributed_client), node_id)); in PYBIND11_MODULE()
|