Home
last modified time | relevance | path

Searched defs:GetDefaultDeviceAssignment (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtpu_client.cc100 StatusOr<DeviceAssignment> PjRtTpuClient::GetDefaultDeviceAssignment( in GetDefaultDeviceAssignment() function in xla::__anon240436b40111::PjRtTpuClient
Dgpu_device.cc47 xla::StatusOr<xla::DeviceAssignment> GpuClient::GetDefaultDeviceAssignment( in GetDefaultDeviceAssignment() function in xla::__anon0c1acb180111::GpuClient
Dpjrt_stream_executor_client.cc254 StatusOr<DeviceAssignment> PjRtStreamExecutorClient::GetDefaultDeviceAssignment( in GetDefaultDeviceAssignment() function in xla::PjRtStreamExecutorClient
/external/tensorflow/tensorflow/compiler/xla/python/
Dpy_client.cc69 PyClient::GetDefaultDeviceAssignment(int num_replicas, int num_partitions) { in GetDefaultDeviceAssignment() function in xla::PyClient
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.cc133 StatusOr<DeviceAssignment> PyTpuClient::GetDefaultDeviceAssignment( in GetDefaultDeviceAssignment() function in xla::PyTpuClient