Searched refs:id_to_device_ (Results 1 – 4 of 4) sorted by relevance
145 auto it = id_to_device_.find(device_id); in LookupDevice()146 if (it != id_to_device_.end()) { in LookupDevice()260 std::map<int, PjRtDevice*> id_to_device_; variable
236 CHECK(id_to_device_.insert({device->id(), device.get()}).second) in PjRtStreamExecutorClient()
116 return id_to_device_; in id_to_device()140 std::map<int, std::shared_ptr<PjRtDevice>> id_to_device_; variable
98 CHECK(id_to_device_.insert({device->id(), device}).second) in PyTpuClient()