keras/src/layers/rnn/time_distributed.py/TimeDistributed/compute_output_shape
class TimeDistributed:    def compute_output_shape(self, input_shape):
        child_input_shape = self._get_child_input_shape(input_shape)
        child_output_shape = self.layer.compute_output_shape(child_input_shape)
        return (child_output_shape[0], input_shape[1], *child_output_shape[1:])