Method Run

tensorflow/core/distributed_runtime/eager/remote_copy_node.cc:316–342

}

Status RemoteCopyNode::Run() {
  if (ctx_->UseSendTensorRPC() && send_device_->IsLocal() &&
      !recv_device_->IsLocal()) {
    return StartRemoteSendTensor();
  }
  Status s = StartSend();
  if (!s.ok()) {
    Abort(s);
    return s;
  }

  // StartRecv() takes care of doing the right thing to dst handle.
  // No need to poison it after this point.
  s = StartRecv();
  if (!s.ok() && errors::IsCancelled(s)) {
    Status send_status = captured_state_->GetSendStatus();
    if (!send_status.ok()) {
      // In this case, Recv is cancelled because the Send op failed. Return the
      // status of the Send op instead.
      s = send_status;
    }
  }

  src_->Unref();
  ctx_->Unref();
  return s;
}

void RemoteCopyNode::Abort(Status status) {
  captured_state_->dst()->Poison(status);