Support TensorFlow < 2.16 and Keras 2