import tensorflow as tf

      wall_time = (time.time() - start_time) / hparams.n_iters
      examples_per_sec = hparams.n_samples / wall_time

      self.report_benchmark(
          name="eager_train_%s%s_%d" %
          ("gpu" if tf.test.is_gpu_available() else "cpu",
           "_defun" if defun else "", sample_size),
          iters=hparams.n_iters,
          extras={"examples_per_sec": examples_per_sec},
          wall_time=wall_time)
