Searched refs:GetTpuDevices (Results 1 – 3 of 3) sorted by relevance
136 StatusOr<std::vector<std::unique_ptr<PjRtStreamExecutorDevice>>> GetTpuDevices( in GetTpuDevices() function216 GetTpuDevices(client, std::move(local_device_states))); in GetTpuClient()
54 GetTpuDevices(const tpu_driver::SystemInfo& system_info);
51 TpuDevice::GetTpuDevices(const tpu_driver::SystemInfo& system_info) { in GetTpuDevices() function in xla::TpuDevice82 TpuDevice::GetTpuDevices(system_info)); in Get()