src/transformers/models/flaubert/modeling_tf_flaubert.py/TFFlaubertPredLayer/call
class TFFlaubertPredLayer:    def call(self, hidden_states):
        hidden_states = self.input_embeddings(hidden_states, mode="linear")
        hidden_states = hidden_states + self.bias

        return hidden_states