KerasDataGenerator.flow_from_tfrecords