def transformer_small_tpu():
  """TPU-friendly version of transformer_small.

  Returns:
    an hparams object.
  """
  hparams = transformer_small()
  update_hparams_for_tpu(hparams)
  return hparams
TPU-friendly version of transformer_small.

  Returns:
    an hparams object.
  
tensor2tensor/models/transformer.py