Searched refs:recv_device_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_copy_node.cc | 84 recv_device_(recv_device), in RemoteCopyNode()
89 DCHECK(!send_device_->IsLocal() || !recv_device_->IsLocal()); in RemoteCopyNode()
135 op.MutableAttrs()->Set("recv_device", recv_device_->name()); in StartSend()
223 Status status = ctx_->GetClient(recv_device_, &eager_client); in RunRemoteRecv()
225 captured_state_->dst()->PoisonRemote(status, recv_device_, context_view_id); in RunRemoteRecv()
237 captured_state_->dst()->PoisonRemote(status, recv_device_, context_view_id); in RunRemoteRecv()
244 Device* recv_device = recv_device_; in RunRemoteRecv()
274 Device* recv_device = ctx_->CanonicalDevice(recv_device_); in StartRecv()
281 op.SetDevice(recv_device_); in StartRecv()
288 op.MutableAttrs()->Set("recv_device", recv_device_->name()); in StartRecv()
[all …]
|
| D | remote_copy_node.h | 77 strings::StrAppend(&out, ", recv_device: ", recv_device_->name()); in DebugString()
167 Device* const recv_device_; variable
|