src/transformers/models/xlm_roberta/modeling_flax_xlm_roberta.py/FlaxXLMRobertaOutput/__call__
class FlaxXLMRobertaOutput:    def __call__(self, hidden_states, attention_output, deterministic: bool = True):
        hidden_states = self.dense(hidden_states)
        hidden_states = self.dropout(hidden_states, deterministic=deterministic)
        hidden_states = self.LayerNorm(hidden_states + attention_output)
        return hidden_states