| 98 | } |
| 99 | |
| 100 | void CollectiveRemoteAccessLocal::PostToPeer( |
| 101 | const string& peer_device, const string& peer_task, const string& key, |
| 102 | Device* from_device, DeviceContext* from_device_ctx, |
| 103 | const AllocatorAttributes& from_alloc_attr, const Tensor* from_tensor, |
| 104 | const DeviceLocality& client_locality, const StatusCallback& done) { |
| 105 | VLOG(1) << "PostToPeer " << this << " key " << key |
| 106 | << " step_id_=" << step_id_; |
| 107 | buf_rendezvous_.ProvideBuf(key, from_device, from_device_ctx, from_tensor, |
| 108 | from_alloc_attr, done); |
| 109 | } |
| 110 | |
| 111 | /*static*/ |
| 112 | void CollectiveRemoteAccessLocal::MemCpyAsync( |