Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtpu_client.cc104 StatusOr<DeviceAssignment> PjRtTpuClient::GetDefaultDeviceAssignment( in GetDefaultDeviceAssignment() function in xla::PjRtTpuClient
Dgpu_device.cc187 xla::StatusOr<xla::DeviceAssignment> GpuClient::GetDefaultDeviceAssignment( in GetDefaultDeviceAssignment() function in xla::__anon56d2c41d0111::GpuClient
Dtfrt_cpu_pjrt_client.cc186 StatusOr<DeviceAssignment> TfrtCpuClient::GetDefaultDeviceAssignment( in GetDefaultDeviceAssignment() function in xla::TfrtCpuClient
Dpjrt_stream_executor_client.cc247 StatusOr<DeviceAssignment> PjRtStreamExecutorClient::GetDefaultDeviceAssignment( in GetDefaultDeviceAssignment() function in xla::PjRtStreamExecutorClient
/external/tensorflow/tensorflow/compiler/xla/python/
Dpy_client.cc123 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.cc134 StatusOr<DeviceAssignment> PyTpuClient::GetDefaultDeviceAssignment( in GetDefaultDeviceAssignment() function in xla::PyTpuClient