Can I use dataset.batch() twice for TimeDistributed?

Question

base_model = VGG16(weights=None, include_top=False, input_shape=(96,96,3))
#base_model.trainable = False

inputs = Input(shape=(seq_len,96,96,3))

x = TimeDistributed(ba