src/transformers/models/mobilevit/modeling_tf_mobilevit.py/TFMobileViTTransformer/call
class TFMobileViTTransformer:    def call(self, hidden_states: tf.Tensor, training: bool = False) -> tf.Tensor:
        for layer_module in self.layers:
            hidden_states = layer_module(hidden_states, training=training)
        return hidden_states