Searched refs:TpuDeviceState (Results 1 – 1 of 1) sorted by relevance
49 class TpuDeviceState : public LocalDeviceState { class51 TpuDeviceState(se::StreamExecutor* executor, LocalClient* client,60 TpuDeviceState::TpuDeviceState(se::StreamExecutor* executor, in TpuDeviceState() function in xla::__anon240436b40111::TpuDeviceState66 Status TpuDeviceState::ThenMemcpyDeviceToDevice( in ThenMemcpyDeviceToDevice()212 absl::make_unique<TpuDeviceState>(executor, client, asynchronous)); in GetTpuClient()