TensorFlow 2.2+ and Custom Training Logic