src/transformers/models/convbert/modeling_tf_convbert.py/TFConvBertPredictionHeadTransform/call
class TFConvBertPredictionHeadTransform:    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