src/transformers/models/luke/modeling_luke.py/EntityPredictionHead/forward
class EntityPredictionHead:    def forward(self, hidden_states):
        hidden_states = self.transform(hidden_states)
        hidden_states = self.decoder(hidden_states) + self.bias

        return hidden_states