Migrating tf.summary usage to TF 2.x | TensorBoard | TensorFlow