Searched refs:drivers_ (Results 1 – 1 of 1) sorted by relevance
270 drivers_.insert({worker_count, std::move(tpu_driver)}); in PodTpuDriver()273 drivers_.insert({driver_info.host_id(), std::move(tpu_driver)}); in PodTpuDriver()283 drivers_[driver_num]->QuerySystemInfo(&driver_info); in PodTpuDriver()312 {current_core, drivers_[tpu_chip.host_id()].get()}); in PodTpuDriver()345 for (auto& driver : drivers_) { in Reset()578 for (int i = 0; i < drivers_.size(); ++i) { in CompileProgram()580 drivers_[i]->CompileProgram(source, num_replicas, {}); in CompileProgram()715 return drivers_[0]->GetLinearizer(); in GetLinearizer()813 absl::flat_hash_map<int32_t, std::unique_ptr<TpuDriver>> drivers_; member in tpu_driver::__anon6b014a980111::PodTpuDriver