class_docstr: Callback for Keras models.
code: def on_epoch_end(self, epoch, logs=None):
      epoch_run_time = time.time() - self.epoch_start
      self.epoch_runtime_log.append(epoch_run_time)
      tf.compat.v1.logging.info(
          "BenchmarkMetric: {'epoch':%d, 'time_taken': %f}"
          % (epoch, epoch_run_time)
      )
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import time
  import tensorflow
  from tensorflow.core.protobuf import rewriter_config_pb2
  from tensorflow.python import tf2
  from tensorflow.python.eager import profiler
label_desc: Use image node_id 6 for calling the TimeHistory obj's underlying member method code with example usage: obj.on_epoch_end(epoch, logs) without return types