docs: def get_examples_per_second_hook(every_n_steps=100,
                                   batch_size=128,
                                   warm_steps=5,
                                   **kwargs):  # pylint: disable=unused-argument
    """Function to get ExamplesPerSecondHook.
  
    Args:
      every_n_steps: `int`, print current and average examples per second every
        N steps.
      batch_size: `int`, total batch size used to calculate examples/second from
        global time.
      warm_steps: skip this number of steps before logging and running average.
      **kwargs: a dictionary of arguments to ExamplesPerSecondHook.
  
    Returns:
      Returns a ProfilerHook that writes out timelines that can be loaded into
      profiling tools like chrome://tracing.
    """
    return hooks.ExamplesPerSecondHook(every_n_steps=every_n_steps,
                                       batch_size=batch_size,
                                       warm_steps=warm_steps)