Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dtpu_client.cc120 StatusOr<DeviceAssignment> PjRtTpuClient::GetDefaultDeviceAssignment( in GetDefaultDeviceAssignment() function in xla::PjRtTpuClient
H A Dpjrt_c_api_client.h145 StatusOr<DeviceAssignment> GetDefaultDeviceAssignment( in GetDefaultDeviceAssignment() function
H A Dgpu_device.cc138 xla::StatusOr<xla::DeviceAssignment> GpuClient::GetDefaultDeviceAssignment( in GetDefaultDeviceAssignment() function in xla::__anonb78c89a70111::GpuClient
H A Dpjrt_client.h489 virtual StatusOr<DeviceAssignment> GetDefaultDeviceAssignment( in GetDefaultDeviceAssignment() function
H A Dtfrt_cpu_pjrt_client.cc232 StatusOr<DeviceAssignment> TfrtCpuClient::GetDefaultDeviceAssignment( in GetDefaultDeviceAssignment() function in xla::TfrtCpuClient
H A Dpjrt_stream_executor_client.cc257 StatusOr<DeviceAssignment> PjRtStreamExecutorClient::GetDefaultDeviceAssignment( in GetDefaultDeviceAssignment() function in xla::PjRtStreamExecutorClient
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dpy_client.cc181 PyClient::GetDefaultDeviceAssignment(int num_replicas, int num_partitions) { in GetDefaultDeviceAssignment() function in xla::PyClient
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
H A Dtpu_client.cc144 StatusOr<DeviceAssignment> PyTpuClient::GetDefaultDeviceAssignment( in GetDefaultDeviceAssignment() function in xla::PyTpuClient