Searched refs:core_to_driver_id_ (Results 1 – 1 of 1) sorted by relevance
313 core_to_driver_id_.insert({current_core, tpu_chip.host_id()}); in PodTpuDriver()518 auto src_driver_id = core_to_driver_id_[src_core_id]; in TransferFromDeviceToDevice()519 auto dst_driver_id = core_to_driver_id_[dst_core_id]; in TransferFromDeviceToDevice()613 cph_iter->second[core_to_driver_id_[core_id]].get(), {})}); in LoadProgram()814 absl::flat_hash_map<int32_t, int32_t> core_to_driver_id_; member in tpu_driver::__anon6b014a980111::PodTpuDriver