src/transformers/models/lxmert/modeling_tf_lxmert.py/TFLxmertPredictionHeadTransform/call
class TFLxmertPredictionHeadTransform:    def call(self, hidden_states: tf.Tensor) -> tf.Tensor:
        hidden_states = self.dense(inputs=hidden_states)
        hidden_states = self.transform_act_fn(hidden_states)
        hidden_states = self.LayerNorm(inputs=hidden_states)

        return hidden_states