TODO(vrv): Figure out how to unify the many different functions that generate RendezvousKey, since many of them have to be consistent with each other.
| 177 | // that generate RendezvousKey, since many of them have to be |
| 178 | // consistent with each other. |
| 179 | string GetRendezvousKey(const string& tensor_name, |
| 180 | const DeviceAttributes& device_info, |
| 181 | const FrameAndIter& frame_iter) { |
| 182 | return strings::StrCat(device_info.name(), ";", |
| 183 | strings::FpToString(device_info.incarnation()), ";", |
| 184 | device_info.name(), ";", tensor_name, ";", |
| 185 | frame_iter.frame_id, ":", frame_iter.iter_id); |
| 186 | } |
| 187 | |
| 188 | // TODO: Any better allocate policy? |
| 189 | void AllocateVisibleCpusForSession( |
no test coverage detected