Searched refs:GpuClient (Results 1 – 1 of 1) sorted by relevance
39 class GpuClient : public xla::PjRtStreamExecutorClient { class47 xla::StatusOr<xla::DeviceAssignment> GpuClient::GetDefaultDeviceAssignment( in GetDefaultDeviceAssignment()336 return std::unique_ptr<PjRtClient>(std::make_unique<GpuClient>( in GetGpuClient()