Searched refs:PjRtTpuClient (Results 1 – 1 of 1) sorted by relevance
76 class PjRtTpuClient : public PjRtStreamExecutorClient { class78 PjRtTpuClient(LocalClient* client,91 PjRtTpuClient::PjRtTpuClient( in PjRtTpuClient() function in xla::__anon240436b40111::PjRtTpuClient100 StatusOr<DeviceAssignment> PjRtTpuClient::GetDefaultDeviceAssignment( in GetDefaultDeviceAssignment()115 StatusOr<absl::optional<std::string>> PjRtTpuClient::ExecutableFingerprint( in ExecutableFingerprint()220 absl::make_unique<PjRtTpuClient>(client, std::move(devices), task_id)); in GetTpuClient()