message: tf.identity should use backing_device so that the output tensor has memory backed on the correct device.
diff: @@ -69,7 +69,7 @@ def identity(input, name=None): # pylint: disable=redefined-builtin
  """
  if context.executing_eagerly() and not hasattr(input, "graph"):
  input = ops.convert_to_tensor(input)
  - in_device = input.device
  + in_device = input.backing_device
  # TODO(ashankar): Does 'identity' need to invoke execution callbacks?
  context_device = context.context().device_name
  if not context_device:
  