src/transformers/models/xlm/modeling_tf_xlm.py/TFXLMPredLayer/call
class TFXLMPredLayer:    def call(self, hidden_states):
        hidden_states = self.input_embeddings(hidden_states, mode="linear")
        hidden_states = hidden_states + self.bias

        return hidden_states