src/transformers/models/mobilebert/modeling_tf_mobilebert.py/TFMobileBertPredictionHeadTransform/call
class TFMobileBertPredictionHeadTransform:    def call(self, hidden_states):
        hidden_states = self.dense(hidden_states)
        hidden_states = self.transform_act_fn(hidden_states)
        hidden_states = self.LayerNorm(hidden_states)
        return hidden_states