def _get_tf_momentum_var(tf_layer):
  """Returns the name of the momentum parameter for a given tf layer.

  Args:
    tf_layer: String for the name of the tf layer.
  Returns:
    String denoting the parameter name for the momentum parameter.
  """
  return '%s/Momentum' % tf_layer