message: Fix control dependencies in Keras distributed TPU loops.
diff: @@ -334,7 +334,7 @@ def experimental_tpu_test_loop(model,
  (_, outputs, updates, _) = _per_replica_execution_function(
  dist_utils.get_distributed_model(model, mode), mode)
  with ops.control_dependencies([updates]):
  - return outputs
  + return [array_ops.identity(out) for out in outputs]
  test_input_data = iterator.get_next()
  per_replica_outputs = current_strategy.experimental_run_v2(
  @@ -481,7 +481,7 @@ def experimental_tpu_predict_loop(model,
  dist_utils.get_distributed_model(model, mode), mode)
  with ops.control_dependencies([updates]):
  - return outputs
  + return [array_ops.identity(out) for out in outputs]
  # TODO(hongjunchoi): When numpy array is passed as an input to `predict()`
  # use numpy arrays directly to avoid cumulating unnecessary input pipeline
  