| 352 | } |
| 353 | |
| 354 | bool TensorHandle::HasRemoteMirror(Device* d) { |
| 355 | tf_shared_lock l(remote_mirrors_mutex_); |
| 356 | auto mirror = remote_mirrors_.find(d); |
| 357 | if (mirror != remote_mirrors_.end()) { |
| 358 | return true; |
| 359 | } |
| 360 | |
| 361 | auto unshaped_mirror = unshaped_remote_mirrors_.find(d); |
| 362 | if (unshaped_mirror != unshaped_remote_mirrors_.end()) { |
| 363 | return true; |
| 364 | } |
| 365 | |
| 366 | return false; |
| 367 | } |
| 368 | |
| 369 | Status TensorHandle::AddUnshapedRemoteMirror( |
| 370 | std::unique_ptr<UnshapedRemoteTensorHandleData> t, Device* d) { |
no test coverage detected