message: Fix the client-side output devices of a remote function.
diff: @@ -185,15 +185,11 @@ Status AddOpRetvalsToResponse(
  } else {
  for (int i = 0; i < num_retvals; i++) {
  TF_RETURN_IF_ERROR(TensorHandleShape(retvals[i], add_shape_proto_fn()));
  - const bool is_remote = retvals[i]->Type() == TensorHandle::REMOTE;
  if (add_device_fn) {
  - *add_device_fn() =
  - is_remote ? absl::get<Device*>(
  - retvals[i]->DeviceOrHostCPU(*eager_context))
  - ->name()
  - : "";
  + Device* device = absl::get<Device*>(retvals[i]->device());
  + *add_device_fn() = device ? device->name() : "";
  }
  - if (is_remote) {
  + if (retvals[i]->Type() == TensorHandle::REMOTE) {
  retvals[i]->Unref();
  } else {
  const int output_num = output_nums.empty() ? i : output_nums.at(i);
  