src/transformers/models/oneformer/modeling_oneformer.py/OneFormerMLPPredictionHead/forward
class OneFormerMLPPredictionHead:    def forward(self, input: Tensor) -> Tensor:
        return self.layers(input)