How should tfrecord data be processed or read for the keras multi input model