I have a trained keras .h5 model and want to change the batch size, to allow processing multiple samples at inference (using a .tflite model). How is that possible in tf2? Unfortunately, I didnt find anything.