message: [tfgan/tpu] Add loss summaries during TPUGANEstimator training if not on TPU.
diff: @@ -294,9 +294,10 @@ def _get_estimator_spec(
  gan_model, gan_loss, gan_loss_no_reduction, get_eval_metric_ops_fn)
  else: # model_fn_lib.ModeKeys.TRAIN:
  gan_loss = tfgan_tuples.GANLoss(
  - generator_loss=generator_loss_fn(gan_model, add_summaries=False),
  + generator_loss=generator_loss_fn(
  + gan_model, add_summaries=not is_on_tpu),
  discriminator_loss=discriminator_loss_fn(
  - gan_model, add_summaries=False))
  + gan_model, add_summaries=not is_on_tpu))
  # Construct optimizers if arguments were callable. For TPUs, they must be
  # `CrossShardOptimizer`.
  