src/transformers/models/canine/modeling_canine.py/CanineLMPredictionHead/forward
class CanineLMPredictionHead:    def forward(self, hidden_states: Tuple[torch.FloatTensor]) -> torch.FloatTensor:
        hidden_states = self.transform(hidden_states)
        hidden_states = self.decoder(hidden_states)
        return hidden_states