Home
last modified time | relevance | path

Searched defs:distributed_client (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dgpu_device.cc339 std::shared_ptr<DistributedRuntimeClient> distributed_client, int node_id, in BuildDistributedDevices()
424 std::shared_ptr<DistributedRuntimeClient> distributed_client, int node_id, in GetGpuClient()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dxla.cc314 -> StatusOr<std::shared_ptr<PyClient>> { in PYBIND11_MODULE()
H A Dxla_client.py68 def make_gpu_client(distributed_client=None, node_id=0, platform_name=None, argument