message: Add the decorator tpu_function.on_device_training_loop to the eval method as well.
diff: @@ -3132,6 +3132,7 @@ def _eval_on_tpu_system(ctx, model_fn_wrapper, dequeue_fn):
  (single_tpu_eval_step, host_calls, captured_scaffold_fn, captured_eval_hooks
  ) = model_fn_wrapper.convert_to_single_tpu_eval_step(dequeue_fn)
  + @tpu_function.on_device_training_loop
  def multi_tpu_eval_steps_on_single_shard():
  return training_loop.repeat(iterations_per_loop_var, single_tpu_eval_step,
  [_ZERO_LOSS])
  