def Owner(owner_name):
  """Sets the owner attribute on a decorated method or class."""

  def _Wrapper(func_or_class):
    """Sets the benchmark owner attribute."""
    func_or_class.__benchmark__owner__ = owner_name
    return func_or_class

  return _Wrapper