message: Update comment to match the code change from cl/321607029
diff: @@ -130,8 +130,7 @@ class Embedding(Layer):
  # since it knows all kernels using the variable only exist on CPU.
  # When eager execution is enabled, the placement decision has to be made
  # right now. Checking for the presence of GPUs to avoid complicating the
  - # TPU codepaths which can handle sparse optimizers. But if we are within
  - # a tf.function, we go back the graph mode logic and rely on the placer.
  + # TPU codepaths which can handle sparse optimizers.
  if context.executing_eagerly() and context.context().num_gpus():
  with ops.device('cpu:0'):
  self.embeddings = self.add_weight(
  